Searched refs:node_to_index_map (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | single_threaded_executor.cc | 105 absl::flat_hash_map<Node*, size_t> node_to_index_map; in Initialize() local
150 node_to_index_map[n] = kernel_index; in Initialize()
179 kernels_[node_to_index_map[e->dst()]].input_start_index + in Initialize()
198 kernels_[node_to_index_map[e->dst()]].input_start_index + in Initialize()
216 kernels_[node_to_index_map[e->dst()]].input_start_index + in Initialize()
|
| /external/executorch/backends/apple/coreml/runtime/inmemoryfs/ |
| D | inmemory_filesystem.cpp | 315 std::unordered_map<InMemoryFileSystem::InMemoryNode*, size_t>& node_to_index_map, in populate() argument
331 node_to_index_map[node] = index; in populate()
337 populate(item.get(), node_to_index_map, alignment, result); in populate()
338 … flattened_node_metadata.child_name_to_indices_map[key] = node_to_index_map[item.get()]; in populate()
340 node_to_index_map[node] = result.size(); in populate()
350 std::unordered_map<InMemoryFileSystem::InMemoryNode*, size_t> node_to_index_map; in flatten() local
352 populate(node, node_to_index_map, alignment, result); in flatten()
|