Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dinference_context.cc389 } else if (graph_ids_to_shared_buffer_tensors_.find(tensor_id) != in GetTensor()
390 graph_ids_to_shared_buffer_tensors_.end()) { in GetTensor()
392 [graph_ids_to_shared_buffer_tensors_[tensor_id]]; in GetTensor()
491 graph_ids_to_shared_buffer_tensors_[usage.first] = in AllocateMemoryForBuffers()
531 const int tensor_index = graph_ids_to_shared_buffer_tensors_[tensor_id]; in AllocateMemoryForBuffers()
Dinference_context.h216 std::map<ValueId, int> graph_ids_to_shared_buffer_tensors_; variable
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dinference_context.cc538 graph_ids_to_shared_buffer_tensors_[usage.first] = in AllocateMemoryForBuffers()
561 const int tensor_index = graph_ids_to_shared_buffer_tensors_[t.first]; in AllocateMemoryForBuffers()
705 } else if (graph_ids_to_shared_buffer_tensors_.find(id) != in GetTensor()
706 graph_ids_to_shared_buffer_tensors_.end()) { in GetTensor()
707 return &shared_buffer_tensors_[graph_ids_to_shared_buffer_tensors_[id]]; in GetTensor()
Dinference_context.h227 std::map<ValueId, int> graph_ids_to_shared_buffer_tensors_; variable