Searched refs:tensor_to_value (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| 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()
|
| D | object_reader.h | 37 TfLiteContext* context, absl::flat_hash_map<int, Value*>* tensor_to_value,
43 absl::flat_hash_map<int, Value*>* tensor_to_value,
48 tensor_to_value_(tensor_to_value), in graph_()
|
| D | model_builder.cc | 2546 absl::flat_hash_map<int, Value*>* tensor_to_value) { in PrecreateIOTensors() argument
2552 context, tensor_to_value, quant_conversion_map, graph, tensor_index)); in PrecreateIOTensors()
2624 absl::flat_hash_map<int, Value*> tensor_to_value; in BuildModel() local
2628 quant_conversion_map, &tensor_to_value)); in BuildModel()
2631 quant_conversion_map, &tensor_to_value)); in BuildModel()
2638 ObjectReader reader(graph, context, tflite_node, &tensor_to_value, in BuildModel()
|