Skip to content

Commit 794d0da

Browse files
committed
Update on "Support named_data in flat_tensor"
Currently flat_tensor ndm only accounts for tensors in get_data, get_num_keys, get_key functions. Add support to return named_data values as well. TODO:consolidate tensors and named_data into one structure in the flatbuffer. This will simplify all the serialization and runtime code. Differential Revision: [D73679683](https://our.internmc.facebook.com/intern/diff/D73679683/) [ghstack-poisoned]
1 parent 2444ee0 commit 794d0da

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

extension/flat_tensor/flat_tensor_data_map.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -253,7 +253,8 @@ ET_NODISCARD Result<const char*> FlatTensorDataMap::get_key(
253253
index,
254254
num_keys);
255255

256-
if (flat_tensor_->named_data() != nullptr && index < flat_tensor_->named_data()->size()) {
256+
if (flat_tensor_->named_data() != nullptr &&
257+
index < flat_tensor_->named_data()->size()) {
257258
return flat_tensor_->named_data()->Get(index)->key()->c_str();
258259
} else {
259260
if (flat_tensor_->named_data() != nullptr) {

0 commit comments

Comments
 (0)