Searched refs:tensor_to_value (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| H A D | object_reader.cc | 36 TfLiteContext* context, absl::flat_hash_map<int, Value*>* tensor_to_value, in ReadNonConstantTensor() argument
44 if (tensor_to_value->find(tensor_idx) == tensor_to_value->end()) { in ReadNonConstantTensor()
82 (*tensor_to_value)[fp_tensor_index] = value; in ReadNonConstantTensor()
94 (*tensor_to_value)[tensor_idx] = value; in ReadNonConstantTensor()
99 *value = (*tensor_to_value)[tensor_idx]; in ReadNonConstantTensor()
|
| H A D | object_reader.h | 40 TfLiteContext* context, absl::flat_hash_map<int, Value*>* tensor_to_value,
46 absl::flat_hash_map<int, Value*>* tensor_to_value,
51 tensor_to_value_(tensor_to_value), in graph_()
|
| H A D | model_builder.cc | 3230 absl::flat_hash_map<int, Value*>* tensor_to_value) { in PrecreateIOTensors() argument
3235 context, tensor_to_value, quant_conversion_map, graph, id)); in PrecreateIOTensors()
3326 absl::flat_hash_map<int, Value*> tensor_to_value; in BuildModelEnforceIO() local
3330 quant_conversion_map, &tensor_to_value)); in BuildModelEnforceIO()
3332 quant_conversion_map, &tensor_to_value)); in BuildModelEnforceIO()
3339 ObjectReader reader(graph, context, tflite_node, &tensor_to_value, in BuildModelEnforceIO()
|