Home
last modified time | relevance | path

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

123456

/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()
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
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
/external/XNNPACK/test/
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
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
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
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
Dtanh-operator-tester.h75 inline TanhOperatorTester& input_scale(float input_scale) { in input_scale() function
82 inline float input_scale() const { in input_scale() function
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
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
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
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
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
Dleaky-relu-operator-tester.h93 inline LeakyReLUOperatorTester& input_scale(float input_scale) { in input_scale() function
100 inline float input_scale() const { in input_scale() function
Dsigmoid-operator-tester.h79 inline SigmoidOperatorTester& input_scale(float input_scale) { in input_scale() argument
86 inline float input_scale() const { in input_scale() function
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
Delu.cc28 const float input_scale = scale_dist(rng); in TEST_F() local
108 const float input_scale = scale_dist(rng); in TEST_F() local
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
/external/XNNPACK/src/operators/
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()
/external/pytorch/aten/src/ATen/native/cuda/
DActivationEluKernel.cu26 const Scalar& input_scale) { in elu_kernel()
51 const Scalar& input_scale, in elu_backward_kernel()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
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
/external/executorch/backends/cadence/reference/operators/
Dquantized_layer_norm.cpp29 double input_scale, in quantized_layer_norm_per_tensor_()
99 float input_scale = in_scale.const_data_ptr<float>()[0]; in quantized_layer_norm_() local
/external/pytorch/aten/src/ATen/native/quantized/cpu/
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()
Dqelu.cpp19 …, int64_t output_zero_point, const Scalar& alpha, const Scalar& scale, const Scalar& input_scale) { in quantized_elu()
/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
Dsoftargmax.c21 float input_scale, in pytorch_qnnp_create_softargmax_nc_q8()
/external/tensorflow/tensorflow/lite/delegates/xnnpack/
Ddequantize_tester.h56 inline DequantizeTester& InputScale(float input_scale) { in InputScale()

123456