Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
H A Dinference_context.cc333 } else if (graph_ids_to_shared_buffer_tensors_.find(tensor_id) != in GetTensor()
334 graph_ids_to_shared_buffer_tensors_.end()) { in GetTensor()
336 [graph_ids_to_shared_buffer_tensors_[tensor_id]]; in GetTensor()
508 graph_ids_to_shared_buffer_tensors_[usage.first] = in AllocateMemoryForBuffers()
565 const int tensor_index = graph_ids_to_shared_buffer_tensors_[tensor_id]; in AllocateMemoryForBuffers()
H A Dinference_context.h179 std::map<ValueId, int> graph_ids_to_shared_buffer_tensors_; variable
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
H A Dinference_context.cc490 &graph_ids_to_shared_buffer_tensors_, &buffer_assignment, in AllocateBufferBasedTensors()
569 const int tensor_index = graph_ids_to_shared_buffer_tensors_[tensor_id]; in AllocateBufferBasedTensors()
894 } else if (graph_ids_to_shared_buffer_tensors_.find(id) != in GetTensor()
895 graph_ids_to_shared_buffer_tensors_.end()) { in GetTensor()
896 return &shared_buffer_tensors_[graph_ids_to_shared_buffer_tensors_[id]]; in GetTensor()
H A Dinference_context.h190 std::map<ValueId, int> graph_ids_to_shared_buffer_tensors_; variable