Home
last modified time | relevance | path

Searched refs:variable_ids_and_refs_ (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dinference_context.cc245 variable_ids_and_refs_[variable_input->id] = variable_input->tensor.ref; in CopyInAndOutIds()
476 } else if (variable_ids_and_refs_.find(id) != variable_ids_and_refs_.end()) { in GetTensorMemoryType()
506 for (auto value_and_ref_value : variable_ids_and_refs_) { in AllocateMemoryForVariableTensors()
703 } else if (variable_ids_and_refs_.find(id) != variable_ids_and_refs_.end()) { in GetTensor()
704 return &variable_tensors_[variable_ids_and_refs_[id]]; in GetTensor()
Dinference_context.h233 std::map<ValueId, ValueId> variable_ids_and_refs_; variable
Dserialization.cc994 for (auto& pair : inference.variable_ids_and_refs_) { in Encode()
1057 inference->variable_ids_and_refs_[variable_id->first()] = in Decode()