Home
last modified time | relevance | path

Searched refs:quant_conversion_map (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Dquantization_util.cc34 const absl::flat_hash_map<int, int>& quant_conversion_map) { in DequantizeInput() argument
35 if (quant_conversion_map.find(input_index) == quant_conversion_map.end()) { in DequantizeInput()
38 int original_tensor_idx = quant_conversion_map.at(input_index); in DequantizeInput()
61 const absl::flat_hash_map<int, int>& quant_conversion_map) { in QuantizeOutput() argument
62 if (quant_conversion_map.find(output_index) == quant_conversion_map.end()) { in QuantizeOutput()
65 int original_tensor_idx = quant_conversion_map.at(output_index); in QuantizeOutput()
91 const absl::flat_hash_map<int, int>& quant_conversion_map) { in DequantizeInputs() argument
93 DequantizeInput(context, static_cast<int>(index), quant_conversion_map); in DequantizeInputs()
100 const absl::flat_hash_map<int, int>& quant_conversion_map) { in DequantizeInputs() argument
102 DequantizeInput(context, static_cast<int>(index), quant_conversion_map); in DequantizeInputs()
[all …]
Dquantization_util.h37 const absl::flat_hash_map<int, int>& quant_conversion_map);
41 const absl::flat_hash_map<int, int>& quant_conversion_map);
50 const absl::flat_hash_map<int, int>& quant_conversion_map);
54 const absl::flat_hash_map<int, int>& quant_conversion_map);
Dquantization_util_test.cc164 absl::flat_hash_map<int, int> quant_conversion_map = {{1, 0}}; in TEST() local
166 auto status = DequantizeInputs(&context, input_indices, quant_conversion_map); in TEST()
189 absl::flat_hash_map<int, int> quant_conversion_map = {{1, 0}}; in TEST() local
191 auto status = DequantizeInputs(&context, input_indices, quant_conversion_map); in TEST()
212 absl::flat_hash_map<int, int> quant_conversion_map = {{0, 1}}; in TEST() local
214 auto status = QuantizeOutputs(&context, output_indices, quant_conversion_map); in TEST()
234 absl::flat_hash_map<int, int> quant_conversion_map = {{0, 1}}; in TEST() local
236 auto status = QuantizeOutputs(&context, output_indices, quant_conversion_map); in TEST()
Dobject_reader.cc37 absl::flat_hash_map<int, int>* quant_conversion_map, GraphFloat32* graph, in ReadNonConstantTensor() argument
53 quant_conversion_map) { in ReadNonConstantTensor()
55 if (quant_conversion_map->find(tensor_idx) == in ReadNonConstantTensor()
56 quant_conversion_map->end()) { in ReadNonConstantTensor()
71 (*quant_conversion_map)[fp_tensor_index] = tensor_idx; in ReadNonConstantTensor()
72 (*quant_conversion_map)[tensor_idx] = fp_tensor_index; in ReadNonConstantTensor()
86 tensor_idx = quant_conversion_map->at(tensor_idx); in ReadNonConstantTensor()
Dobject_reader.h38 absl::flat_hash_map<int, int>* quant_conversion_map, GraphFloat32* graph,
44 absl::flat_hash_map<int, int>* quant_conversion_map = nullptr)
49 quant_conversion_map_(quant_conversion_map) {} in graph_()
Dmodel_builder.h50 absl::flat_hash_map<int, int>* quant_conversion_map = nullptr);
64 absl::flat_hash_map<int, int>* quant_conversion_map = nullptr);
Dmodel_builder.cc2545 absl::flat_hash_map<int, int>* quant_conversion_map, in PrecreateIOTensors() argument
2552 context, tensor_to_value, quant_conversion_map, graph, tensor_index)); in PrecreateIOTensors()
2598 absl::flat_hash_map<int, int>* quant_conversion_map) { in BuildModel() argument
2614 registration, /*allow_quant_ops=*/quant_conversion_map != nullptr); in BuildModel()
2628 quant_conversion_map, &tensor_to_value)); in BuildModel()
2631 quant_conversion_map, &tensor_to_value)); in BuildModel()
2639 quant_conversion_map); in BuildModel()
2669 GraphFloat32* graph, absl::flat_hash_map<int, int>* quant_conversion_map) { in BuildFinalModel() argument
2671 BuildModel(context, delegate_params, graph, quant_conversion_map)); in BuildFinalModel()