/external/tensorflow/tensorflow/lite/kernels/internal/ |
D | logsoftmax_quantized_test.cc | 39 int32 input_offset, const double input_scale, in RunLogSoftmaxFloatReference() 72 int32 input_offset, const double input_scale, in RunLogSoftmaxFloatReference() 148 const double input_scale, int stride, float beta) { in RunOneLogSoftmaxTest() 200 const double input_scale, int stride, float beta) { in RunOneLogSoftmaxTest() 254 const double input_scale = std::pow(10.0, UniformRandomFloat(-2.0, 1.0)); in TryOneUniformLogSoftmax() local 287 const double input_scale = std::pow(10.0, UniformRandomFloat(-2.0, 1.0)); in TryOneSkyscraperLogSoftmax() local
|
D | softmax_quantized_test.cc | 37 int32 input_offset, const double input_scale, in RunSoftmaxFloatReference() 103 const double input_scale, int stride, float beta) { in RunOneSoftmaxTest() 158 const double input_scale = std::pow(10.0, UniformRandomFloat(-2.0, 1.0)); in TryOneUniformSoftmax() local 191 const double input_scale = std::pow(10.0, UniformRandomFloat(-2.0, 1.0)); in TryOneSkyscraperSoftmax() local
|
D | quantization_util.cc | 254 void PreprocessSoftmaxScaling(double beta, double input_scale, in PreprocessSoftmaxScaling() 285 void PreprocessLogSoftmaxScalingExp(double beta, double input_scale, in PreprocessLogSoftmaxScalingExp()
|
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() 116 xla::XlaOp input_scale = in Compile() local 206 xla::XlaOp nudged_input_min, nudged_input_max, input_scale; in Compile() local 249 xla::XlaOp nudged_input_min, nudged_input_max, input_scale; in Compile() local
|
/external/tensorflow/tensorflow/lite/kernels/ |
D | kernel_util.cc | 72 const float input_scale = input->params.scale; in PopulateConvolutionQuantizationParams() local
|
/external/tensorflow/tensorflow/core/kernels/ |
D | quantized_instance_norm.cc | 279 float input_scale = (input_max - input_min) / 255.0f; in Compute() local
|
/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
D | reference_ops.h | 1657 const float* input_scale = params.input_scale; in ConcatenationWithScaling() local 1782 const float* input_scale = params.input_scale; in PackWithScaling() local 3650 float input_scale, const RuntimeShape& unextended_output_shape, in Mean() 3711 float input_scale, const int* input_dims, in QuantizedMeanOrSum()
|
D | legacy_reference_ops.h | 621 const float* input_scale, int inputs_count, in Concatenation() 1080 const float* input_scale, int inputs_count, Scalar* output_data, 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() 1920 float input_scale, const RuntimeShape& unextended_output_shape, in Mean()
|