Home
last modified time | relevance | path

Searched refs:const_tensors_descs_ (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/metal/
Dinference_context.cc225 const_tensors_descs_[outputs[0]->id] = in Compile()
227 const_tensors_descs_[outputs[0]->id].UploadData(attr.tensor); in Compile()
470 for (auto& description : const_tensors_descs_) { in AllocateMemoryForConstTensors()
474 const_tensors_descs_.clear(); in AllocateMemoryForConstTensors()
Dinference_context.h213 std::map<ValueId, TensorDescriptor> const_tensors_descs_; variable
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dinference_context.cc317 const_tensors_descs_[outputs[0]->id] = in ConvertOperations()
319 const_tensors_descs_[outputs[0]->id].UploadData(attr.tensor); in ConvertOperations()
495 for (auto& description : const_tensors_descs_) { in AllocateMemoryForConstTensors()
735 const_tensors_descs_.clear(); in ReleaseCPURepresentation()
Dinference_context.h220 absl::flat_hash_map<ValueId, TensorDescriptor> const_tensors_descs_; variable
Dserialization.cc986 for (const auto& tensor : inference.const_tensors_descs_) { in Encode()
1047 inference->const_tensors_descs_[tensor_fb->id()] = std::move(desc); in Decode()