Home
last modified time | relevance | path

Searched refs:external_tensor_to_nodes_ (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
H A Dinference_context.h175 absl::flat_hash_map<ValueId, std::vector<int>> external_tensor_to_nodes_; variable
H A Dinference_context.cc778 for (int node_index : external_tensor_to_nodes_[tensor_id]) { in SetTensor()
811 external_tensor_to_nodes_[external.first].push_back(i); in PrepareExternal()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
H A Dinference_context.h178 absl::flat_hash_map<ValueId, std::vector<int>> external_tensor_to_nodes_; variable
H A Dinference_context.cc706 for (int node_index : external_tensor_to_nodes_[tensor_id]) { in SetTensor()
739 external_tensor_to_nodes_[external.first].push_back(i); in PrepareExternal()