Home
last modified time | relevance | path

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

123456

/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/
H A 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()
206 const double input_scale, int stride, float beta) { in RunOneLogSoftmaxTest()
260 const double input_scale = std::pow(10.0, UniformRandomFloat(-2.0, 1.0)); in TryOneUniformLogSoftmax() local
293 const double input_scale = std::pow(10.0, UniformRandomFloat(-2.0, 1.0)); in TryOneSkyscraperLogSoftmax() local
H A Dsoftmax_quantized_test.cc37 int32 input_offset, const double input_scale, in RunSoftmaxFloatReference()
108 const double input_scale, int stride, float beta) { in RunOneSoftmaxTest()
182 const double input_scale = std::pow(10.0, UniformRandomFloat(-2.0, 1.0)); in TryOneUniformSoftmax() local
215 const double input_scale = std::pow(10.0, UniformRandomFloat(-2.0, 1.0)); in TryOneSkyscraperSoftmax() local
/aosp_15_r20/external/XNNPACK/src/operators/
H A Dlut-elementwise-nc.c25 float input_scale, in create_lut_elementwise_nc()
149 float input_scale, in xnn_create_elu_nc_qs8()
183 float input_scale, in xnn_create_sigmoid_nc_qs8()
220 float input_scale, in xnn_create_sigmoid_nc_qu8()
261 float input_scale, in xnn_create_tanh_nc_qs8()
298 float input_scale, in xnn_create_tanh_nc_qu8()
/aosp_15_r20/external/XNNPACK/test/
H A Dtanh-nc.cc45 for (float input_scale = 1.0e-2f; input_scale < 1.0e+2f; input_scale *= 10.0f) { in TEST() local
125 for (float input_scale = 1.0e-2f; input_scale < 1.0e+2f; input_scale *= 10.0f) { in TEST() local
189 for (float input_scale = 1.0e-2f; input_scale < 1.0e+2f; input_scale *= 10.0f) { in TEST() local
251 for (float input_scale = 1.0e-2f; input_scale < 1.0e+2f; input_scale *= 10.0f) { in TEST() local
331 for (float input_scale = 1.0e-2f; input_scale < 1.0e+2f; input_scale *= 10.0f) { in TEST() local
395 for (float input_scale = 1.0e-2f; input_scale < 1.0e+2f; input_scale *= 10.0f) { in TEST() local
H A Dsigmoid-nc.cc158 for (float input_scale = 1.0e-2f; input_scale < 1.0e+2f; input_scale *= 10.0f) { in TEST() local
238 for (float input_scale = 1.0e-2f; input_scale < 1.0e+2f; input_scale *= 10.0f) { in TEST() local
302 for (float input_scale = 1.0e-2f; input_scale < 1.0e+2f; input_scale *= 10.0f) { in TEST() local
365 for (float input_scale = 1.0e-2f; input_scale < 1.0e+2f; input_scale *= 10.0f) { in TEST() local
445 for (float input_scale = 1.0e-2f; input_scale < 1.0e+2f; input_scale *= 10.0f) { in TEST() local
509 for (float input_scale = 1.0e-2f; input_scale < 1.0e+2f; input_scale *= 10.0f) { in TEST() local
H A Dsigmoid.cc28 const float input_scale = scale_dist(rng); in TEST_F() local
68 const float input_scale = scale_dist(rng); in TEST_F() local
146 const float input_scale = scale_dist(rng); in TEST_F() local
209 const float input_scale = scale_dist(rng); in TEST_F() local
H A Dleaky-relu.cc69 const float input_scale = scale_dist(rng); in TEST_F() local
110 const float input_scale = scale_dist(rng); in TEST_F() local
212 const float input_scale = scale_dist(rng); in TEST_F() local
278 const float input_scale = scale_dist(rng); in TEST_F() local
H A Dclamp.cc29 const float input_scale = scale_dist(rng); in TEST_F() local
73 const float input_scale = scale_dist(rng); in TEST_F() local
155 const float input_scale = scale_dist(rng); in TEST_F() local
220 const float input_scale = scale_dist(rng); in TEST_F() local
H A Dstatic-transpose.cc48 const float input_scale = scale_dist(rng); in TEST_F() local
96 const float input_scale = scale_dist(rng); in TEST_F() local
186 const float input_scale = scale_dist(rng); in TEST_F() local
251 const float input_scale = scale_dist(rng); in TEST_F() local
H A Ddepth-to-space.cc100 const float input_scale = scale_dist(rng); in TEST_F() local
140 const float input_scale = scale_dist(rng); in TEST_F() local
218 const float input_scale = scale_dist(rng); in TEST_F() local
282 const float input_scale = scale_dist(rng); in TEST_F() local
H A Dconvert.cc142 const float input_scale = std::uniform_real_distribution<float>(0.0883883f, 11.3137f)(rng); in TEST_F() local
221 const float input_scale = std::uniform_real_distribution<float>(0.0883883f, 11.3137f)(rng); in TEST_F() local
437 const float input_scale = std::uniform_real_distribution<float>(0.25f, 4.0f)(rng); in TEST_F() local
563 const float input_scale = std::uniform_real_distribution<float>(0.25f, 4.0f)(rng); in TEST_F() local
H A Dsoftmax-operator-tester.h79 inline SoftMaxOperatorTester& input_scale(float input_scale) { in input_scale() function
86 inline float input_scale() const { in input_scale() function
H A Dtanh-operator-tester.h75 inline TanhOperatorTester& input_scale(float input_scale) { in input_scale() argument
82 inline float input_scale() const { in input_scale() function
H A Delu-operator-tester.h86 inline ELUOperatorTester& input_scale(float input_scale) { in input_scale() function
93 inline float input_scale() const { in input_scale() function
H A Dleaky-relu-operator-tester.h93 inline LeakyReLUOperatorTester& input_scale(float input_scale) { in input_scale() argument
100 inline float input_scale() const { in input_scale() function
H A Dsigmoid-operator-tester.h79 inline SigmoidOperatorTester& input_scale(float input_scale) { in input_scale() function
86 inline float input_scale() const { in input_scale() function
H A Dglobal-average-pooling-operator-tester.h88 inline GlobalAveragePoolingOperatorTester& input_scale(float input_scale) { in input_scale() function
95 inline float input_scale() const { in input_scale() function
H A Delu-nc.cc185 for (float input_scale = 1.0e-2f; input_scale < 1.0e+2f; input_scale *= 10.0f) { in TEST() local
265 for (float input_scale = 1.0e-2f; input_scale < 1.0e+2f; input_scale *= 10.0f) { in TEST() local
342 for (float input_scale = 1.0e-2f; input_scale < 1.0e+2f; input_scale *= 10.0f) { in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dfake_quantize_ops.cc74 const xla::XlaOp& input_scale) { in Quantize()
159 xla::XlaOp nudged_input_min, nudged_input_max, input_scale; in Compile() local
204 xla::XlaOp nudged_input_min, nudged_input_max, input_scale; in Compile() local
273 xla::XlaOp nudged_input_min, nudged_input_max, input_scale; in Compile() local
334 xla::XlaOp nudged_input_min, nudged_input_max, input_scale; in Compile() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dqlinear.cpp291 double input_scale, in apply_with_input_q_dq_qweight_dq_output_fp32()
301 double input_scale, in apply_with_input_q_dq_qweight_dq_relu_output_fp32()
313 double input_scale, in apply_with_input_q_dq_qweight_dq_output_fp32_impl()
472 const auto input_scale = input_contig.q_scale(); in apply_impl_xnnp() local
630 auto input_scale = input_contig.q_scale(); in apply_impl() local
819 double input_scale = input.q_scale(); in apply_impl() local
915 double input_scale, in linear_int8_with_onednn_weight()
1179 double input_scale, in run()
H A Dqlinear_prepack.cpp363 const at::Tensor& input_scale, in wrapped_quantized_linear()
419 [[maybe_unused]] const at::Tensor& input_scale, in wrapped_quantized_linear_meta()
484 …:Tensor _wrapped_quantized_linear_prepacked(const at::Tensor& input, const at::Tensor& input_scale, in _wrapped_quantized_linear_prepacked()
542 [[maybe_unused]] const at::Tensor& input_scale, in _wrapped_quantized_linear_prepacked_meta()
H A DQnnpackUtils.h60 std::optional<double> input_scale; member
305 std::optional<double> input_scale; member
449 const float input_scale, in generate_requantization_scales()
/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/
H A Dquantization_utils_test.cc462 const float input_scale = 0.05; in TEST_F() local
508 const float input_scale = 0.05; in TEST_F() local
756 const float input_scale = 0.5; in TEST_F() local
798 const float input_scale = 0.5; in TEST_F() local
856 const float input_scale = 0.5; in TEST_F() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DActivationEluKernel.cu26 const Scalar& input_scale) { in elu_kernel()
51 const Scalar& input_scale, in elu_backward_kernel()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
H A Dlegalize_utils.cc56 Value input_val, double input_scale, in buildRescaleToInt32()
91 double input_scale = input_qtype.getScale(); in buildRescaleOpConvOutput() local
168 double input_scale, int32_t input_zp, in getTosaConst8bitTable()
242 double input_scale, int32_t input_zp, in getTosaConst32bitTable()

123456