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 | 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_()
|
| D | model_builder.cc | 2736 absl::flat_hash_map<int, Value*>* tensor_to_value) { in PrecreateIOTensors() argument
2741 context, tensor_to_value, quant_conversion_map, graph, id)); in PrecreateIOTensors()
2832 absl::flat_hash_map<int, Value*> tensor_to_value; in BuildModelEnforceIO() local
2836 quant_conversion_map, &tensor_to_value)); in BuildModelEnforceIO()
2838 quant_conversion_map, &tensor_to_value)); in BuildModelEnforceIO()
2845 ObjectReader reader(graph, context, tflite_node, &tensor_to_value, in BuildModelEnforceIO()
|