| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | per_channel_dequantize_test.cc | 32 const int quantized_dimension = 0; in TEST() local
43 op_params.quantized_dimension = quantized_dimension; in TEST()
54 const int quantized_dimension = 2; in TEST() local
65 op_params.quantized_dimension = quantized_dimension; in TEST()
76 const int quantized_dimension = 0; in TEST() local
88 op_params.quantized_dimension = quantized_dimension; in TEST()
100 const int quantized_dimension = 3; in TEST() local
112 op_params.quantized_dimension = quantized_dimension; in TEST()
|
| /external/tensorflow/tensorflow/lite/micro/ |
| D | micro_utils.cc | 37 int quantized_dimension, in SignedSymmetricPerChannelQuantize() argument
41 int channel_count = dims->data[quantized_dimension]; in SignedSymmetricPerChannelQuantize()
46 if (quantized_dimension == 0) { in SignedSymmetricPerChannelQuantize()
49 } else if (quantized_dimension == 3) { in SignedSymmetricPerChannelQuantize()
|
| D | test_helpers.cc | 1021 TfLiteAffineQuantization* affine_quant, int quantized_dimension, in CreatePerChannelQuantizedBiasTensor() argument
1024 int num_channels = dims->data[quantized_dimension]; in CreatePerChannelQuantizedBiasTensor()
1039 affine_quant->quantized_dimension = quantized_dimension; in CreatePerChannelQuantizedBiasTensor()
1049 int quantized_dimension, bool is_variable) { in CreateSymmetricPerChannelQuantizedTensor() argument
1050 int channel_count = dims->data[quantized_dimension]; in CreateSymmetricPerChannelQuantizedTensor()
1054 SignedSymmetricPerChannelQuantize(input, dims, quantized_dimension, quantized, in CreateSymmetricPerChannelQuantizedTensor()
1063 affine_quant->quantized_dimension = quantized_dimension; in CreateSymmetricPerChannelQuantizedTensor()
|
| D | test_helpers.h | 212 TfLiteAffineQuantization* affine_quant, int quantized_dimension,
218 int quantized_dimension, bool is_variable = false);
|
| D | micro_utils.h | 96 int quantized_dimension,
|
| D | micro_allocator.cc | 566 quantization->quantized_dimension = src_quantization->quantized_dimension(); in InitializeTfLiteTensorFromFlatbuffer()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | dequantize.h | 57 const int32_t quantized_dimension = op_params.quantized_dimension; in PerChannelDequantize() local
66 const int channel = current_dim[quantized_dimension]; in PerChannelDequantize()
|
| /external/tensorflow/tensorflow/lite/ |
| D | interpreter_builder.cc | 384 if (src_quantization->quantized_dimension() < 0 || in ParseQuantization()
386 src_quantization->quantized_dimension() >= dims.size())) { in ParseQuantization()
389 src_quantization->quantized_dimension()); in ParseQuantization()
397 num_scales != dims[src_quantization->quantized_dimension()])) { in ParseQuantization()
401 dims[src_quantization->quantized_dimension()], num_scales); in ParseQuantization()
416 affine_quantization->quantized_dimension = in ParseQuantization()
417 src_quantization->quantized_dimension(); in ParseQuantization()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/arc_mli/ |
| D | mli_tf_utils.h | 70 mliT->el_params.asym.dim = affine_quantization->quantized_dimension; in ConvertToMliQuantParamsPerChannel()
74 mliT->shape[affine_quantization->quantized_dimension]; in ConvertToMliQuantParamsPerChannel()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | model_builder_helper.h | 89 op_params.quantized_dimension = quant_params->quantized_dimension; in DequantizeConstantTensor()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | kernel_util_test.cc | 310 filter_params->quantized_dimension = 0; in TEST_F()
413 filter_params->quantized_dimension = 0; in TEST_F()
524 filter_params->quantized_dimension = 0; in TEST_F()
628 filter_params->quantized_dimension = 0; in TEST_F()
727 filter_params->quantized_dimension = 0; in TEST_F()
|
| D | kernel_util.cc | 234 filter->dims->data[affine_quantization->quantized_dimension]); in PopulateConvolutionQuantizationParams()
|
| D | test_util.h | 407 const int channel_index = params->quantized_dimension; in PerChannelSymmetricQuantizeAndPopulate()
824 affine_quantization->quantized_dimension = 0; in QuantizeTensor()
|
| D | depthwise_conv.cc | 205 filter->dims->data[affine_quantization->quantized_dimension]); in Prepare()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/ |
| D | importer_test_min_max.cc | 136 output_tensor->quantization->quantized_dimension = shape.size() - 1; in InjectStatsToFullyConnected()
|
| /external/tensorflow/tensorflow/lite/python/interpreter_wrapper/ |
| D | interpreter_wrapper.cc | 392 int32_t quantized_dimension = 0; in TensorQuantizationParameters() local
404 quantized_dimension = q_params->quantized_dimension; in TensorQuantizationParameters()
413 PyTuple_SET_ITEM(result, 2, PyLong_FromLong(quantized_dimension)); in TensorQuantizationParameters()
|
| /external/tensorflow/tensorflow/lite/tools/optimize/ |
| D | quantization_utils.cc | 159 quantization_params->quantized_dimension = channel_dim_index; in FillPerChannelMinMax()
533 int quantized_dimension, in AddQuantizationParams() argument
551 tensor->quantization->quantized_dimension = quantized_dimension; in AddQuantizationParams()
|
| D | quantization_utils.h | 86 int quantized_dimension,
|
| D | quantization_utils_test.cc | 307 EXPECT_EQ(quantization_params.quantized_dimension, channel_dim_idx); in TEST_F()
329 EXPECT_EQ(quantization_params.quantized_dimension, channel_dim_idx); in TEST_F()
|
| D | quantize_model.cc | 1235 if (tensor->quantization->quantized_dimension != in FillQuantizationParams()
1242 tensor->quantization->quantized_dimension); in FillQuantizationParams()
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | interpreter_test.py | 80 quantized_dimension, params): argument
83 self.assertEqual(quantized_dimension, params['quantized_dimension'])
|
| /external/tensorflow/tensorflow/lite/c/ |
| D | common.h | 290 int32_t quantized_dimension; member
|
| /external/tensorflow/tensorflow/lite/g3doc/performance/ |
| D | quantization_spec.md | 45 `zero_point` per slice in the `quantized_dimension`. The quantized dimension
55 Often, the `quantized_dimension` is the `output_channel` of the weights of
|
| /external/tensorflow/tensorflow/lite/tools/serialization/ |
| D | writer_lib.cc | 230 QuantizationDetails_NONE, 0, params->quantized_dimension); in ExportTensors()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_import.cc | 172 quant_params.zero_point, quant_params.quantized_dimension, storage_min, in GetQuantizedType()
274 axis = b.getI64IntegerAttr(tensor.quantization->quantized_dimension); in ConvertMinMaxToStatsOp()
|