| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | logsoftmax_quantized_test.cc | 39 int32 input_offset, const double input_scale, in RunLogSoftmaxFloatReference() argument
50 dq_params.scale = input_scale; in RunLogSoftmaxFloatReference()
72 int32 input_offset, const double input_scale, in RunLogSoftmaxFloatReference() argument
83 dq_params.scale = input_scale; in RunLogSoftmaxFloatReference()
148 const double input_scale, int stride, float beta) { in RunOneLogSoftmaxTest() argument
155 input_scale, stride, beta, in RunOneLogSoftmaxTest()
164 beta, input_scale, kScaledDiffIntegerBits, &input_beta_multiplier, in RunOneLogSoftmaxTest()
200 const double input_scale, int stride, float beta) { in RunOneLogSoftmaxTest() argument
206 input_scale, stride, beta, in RunOneLogSoftmaxTest()
215 beta, input_scale, kScaledDiffIntegerBits, &input_beta_multiplier, in RunOneLogSoftmaxTest()
[all …]
|
| D | softmax_quantized_test.cc | 37 int32 input_offset, const double input_scale, in RunSoftmaxFloatReference() argument
48 dq_params.scale = input_scale; in RunSoftmaxFloatReference()
103 const double input_scale, int stride, float beta) { in RunOneSoftmaxTest() argument
109 RunSoftmaxFloatReference(input_data, shape_common, input_offset, input_scale, in RunOneSoftmaxTest()
115 tflite::PreprocessSoftmaxScaling(beta, input_scale, kScaledDiffIntegerBits, in RunOneSoftmaxTest()
158 const double input_scale = std::pow(10.0, UniformRandomFloat(-2.0, 1.0)); in TryOneUniformSoftmax() local
168 RunOneSoftmaxTest(input_data.data(), shape_common, input_offset, input_scale, in TryOneUniformSoftmax()
191 const double input_scale = std::pow(10.0, UniformRandomFloat(-2.0, 1.0)); in TryOneSkyscraperSoftmax() local
207 RunOneSoftmaxTest(input_data.data(), shape_common, input_offset, input_scale, in TryOneSkyscraperSoftmax()
|
| D | quantization_util.cc | 254 void PreprocessSoftmaxScaling(double beta, double input_scale, in PreprocessSoftmaxScaling() argument
267 const double input_beta = IntegerDoubleMultiply(beta, input_scale); in PreprocessSoftmaxScaling()
278 beta * input_scale * (1 << (31 - input_integer_bits)), (1ll << 31) - 1.0); in PreprocessSoftmaxScaling()
285 void PreprocessLogSoftmaxScalingExp(double beta, double input_scale, in PreprocessLogSoftmaxScalingExp() argument
291 PreprocessSoftmaxScaling(beta, input_scale, input_integer_bits, in PreprocessLogSoftmaxScalingExp()
|
| D | quantization_util.h | 240 void PreprocessSoftmaxScaling(double beta, double input_scale,
244 void PreprocessLogSoftmaxScalingExp(double beta, double input_scale,
|
| D | types.h | 758 const float* input_scale; member
893 const float* input_scale; member
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | fake_quantize_ops.cc | 73 const xla::XlaOp& input_scale) { in Quantize() argument
75 xla::XlaOp inv_scale = xla::Div(one, input_scale); in Quantize()
82 return xla::Add(xla::Mul(rounded, input_scale), nudged_input_min); in Quantize()
116 xla::XlaOp input_scale = in Compile() local
119 nudged_input_max, input_scale); in Compile()
206 xla::XlaOp nudged_input_min, nudged_input_max, input_scale; in Compile() local
208 &nudged_input_min, &nudged_input_max, &input_scale); in Compile()
211 nudged_input_max, input_scale); in Compile()
249 xla::XlaOp nudged_input_min, nudged_input_max, input_scale; in Compile() local
251 &nudged_input_min, &nudged_input_max, &input_scale); in Compile()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | kernel_util.cc | 72 const float input_scale = input->params.scale; in PopulateConvolutionQuantizationParams() local
77 const double effective_output_scale = static_cast<double>(input_scale) * in PopulateConvolutionQuantizationParams()
|
| D | concatenation.cc | 132 op_params.input_scale = all_inputs.scale(); \ in Eval()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | quantized_instance_norm.cc | 279 float input_scale = (input_max - input_min) / 255.0f; in Compute() local
354 (input_scale * in Compute()
361 input_scale * in Compute()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 1657 const float* input_scale = params.input_scale; in ConcatenationWithScaling() local
1694 input_scale[i] == output_scale) { in ConcatenationWithScaling()
1697 const float scale = input_scale[i] * inverse_output_scale; in ConcatenationWithScaling()
1782 const float* input_scale = params.input_scale; in PackWithScaling() local
1802 input_scale[i] == output_scale) { in PackWithScaling()
1807 const float scale = input_scale[i] * inverse_output_scale; in PackWithScaling()
3650 float input_scale, const RuntimeShape& unextended_output_shape, in Mean() argument
3678 (input_zero_point == output_zero_point && input_scale == output_scale); in Mean()
3681 scale = input_scale / output_scale; in Mean()
3711 float input_scale, const int* input_dims, in QuantizedMeanOrSum() argument
[all …]
|
| D | legacy_reference_ops.h | 621 const float* input_scale, int inputs_count, in Concatenation() argument
634 op_params.input_scale = input_scale; in Concatenation()
1080 const float* input_scale, int inputs_count, Scalar* output_data, in Pack() argument
1092 op_params.input_scale = input_scale; in Pack()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| D | types.h | 734 const float* input_scale; member
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 1758 int32 input_zero_point, float input_scale, in MeanImpl() argument
1780 (input_zero_point == output_zero_point && input_scale == output_scale); in MeanImpl()
1783 scale = input_scale / output_scale; in MeanImpl()
1880 int32 input_zero_point, float input_scale, in MeanWorkerTask()
1888 input_scale_(input_scale), in MeanWorkerTask()
1920 float input_scale, const RuntimeShape& unextended_output_shape, in Mean() argument
1950 MeanImpl(op_params, input_shape, input_data, input_zero_point, input_scale, in Mean()
1963 input_zero_point, input_scale, output_shape, in Mean()
|