Home
last modified time | relevance | path

Searched defs:input_scale (Results 1 – 11 of 11) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/
Dlogsoftmax_quantized_test.cc39 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
Dsoftmax_quantized_test.cc37 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
Dquantization_util.cc254 void PreprocessSoftmaxScaling(double beta, double input_scale, in PreprocessSoftmaxScaling()
285 void PreprocessLogSoftmaxScalingExp(double beta, double input_scale, in PreprocessLogSoftmaxScalingExp()
Dtypes.h758 const float* input_scale; member
893 const float* input_scale; member
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dfake_quantize_ops.cc73 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/
Dkernel_util.cc72 const float input_scale = input->params.scale; in PopulateConvolutionQuantizationParams() local
/external/tensorflow/tensorflow/core/kernels/
Dquantized_instance_norm.cc279 float input_scale = (input_max - input_min) / 255.0f; in Compute() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dreference_ops.h1657 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()
Dlegacy_reference_ops.h621 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/
Dtypes.h734 const float* input_scale; member
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Doptimized_ops.h1758 int32 input_zero_point, float input_scale, in MeanImpl()
1920 float input_scale, const RuntimeShape& unextended_output_shape, in Mean()