Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dtensor_desc.cc124 auto it1 = state_vars_.find("ElementsX2"); in GetGPUResources()
125 if (it1 != state_vars_.end() && it1->second == "true") { in GetGPUResources()
128 auto it2 = state_vars_.find("ElementsX4"); in GetGPUResources()
129 if (it2 != state_vars_.end() && it2->second == "true") { in GetGPUResources()
200 state_vars_["batch_id"] = args[0]; in PerformSelector()
517 auto it = state_vars_.find("batch_id"); in PerformGetWHOffsetSelector()
519 if (it == state_vars_.end()) { in PerformGetWHOffsetSelector()
734 auto it = state_vars_.find(read_type); in GetDataTypeFromTemplateArgs()
735 if (it == state_vars_.end()) { in GetDataTypeFromTemplateArgs()
771 auto it1 = state_vars_.find("ElementsX2"); in GetWidthSize()
[all …]
Dgpu_object_desc.h113 state_vars_[key] = value; in SetStateVar()
140 mutable std::map<std::string, std::string> state_vars_; variable
/external/tensorflow/tensorflow/lite/delegates/gpu/cl/
Dserialization.cc335 for (auto& v0 : desc.state_vars_) { in Encode()
355 obj->state_vars_[key] = value; in Decode()