Home
last modified time | relevance | path

Searched defs:output_scale (Results 1 – 25 of 167) sorted by relevance

1234567

/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dqlinear.cpp39 double output_scale, in apply_impl()
240 double output_scale, in apply()
254 double output_scale, in apply_relu()
267 double output_scale, in apply_out()
279 double output_scale, in apply_relu_out()
456 double output_scale, in apply_impl_xnnp()
612 double output_scale, in apply_impl()
763 double output_scale, in apply()
776 double output_scale, in apply_relu()
793 double output_scale, in apply_impl()
[all …]
H A Dqconv.cpp239 double output_scale, in apply()
247 double output_scale, in apply_relu()
256 double output_scale, in apply_impl()
569 const at::Tensor& act, double output_scale, int64_t output_zero_point) { in apply_impl_xnnp()
828 double output_scale, in apply_impl()
1057 double output_scale, in apply()
1071 double output_scale, in apply_relu()
1118 double output_scale, in apply()
1126 double output_scale, in apply_relu()
1135 double output_scale, in apply_add()
[all …]
H A Dqnormalization.cpp29 double output_scale, in quantized_layer_norm_impl()
62 double output_scale, in quantized_group_norm_impl()
115 double output_scale, in quantized_instance_norm_impl()
155 double output_scale, in TORCH_LIBRARY_IMPL()
168 double output_scale, in TORCH_LIBRARY_IMPL()
169 int64_t output_zero_point) { in TORCH_LIBRARY_IMPL()
H A DNormalization.cpp34 double output_scale, in compute_fused_params()
62 double output_scale, in q_batch_norm1d_impl()
170 double output_scale, in q_batch_norm2d_impl()
264 double output_scale, in q_batch_norm3d_impl()
361 double output_scale, in q_batch_norm_impl()
387 double output_scale, in quantized_batch_norm()
H A Dqsigmoid.cpp30 Tensor input, double output_scale, int64_t output_zero_point) { in qnnpack_sigmoid()
104 constexpr double output_scale = 1.0f / 256.0f; in sigmoid_quantized_cpu() local
117 double output_scale = 0.00390625; // 1.0 / 2^8 in sigmoid_quantized_cpu() local
134 static Tensor run(Tensor qx, double output_scale, int64_t output_zero_point) { in run()
H A Dqsoftmax.cpp25 const double output_scale, in is_qnnpack_compatible()
122 const double output_scale, in qsoftmax_naive()
133 const double output_scale, in qsoftmax()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cudnn/
H A DConv.cpp67 …ply_impl_helper(const at::Tensor& quantized_output, const at::Tensor& input, double output_scale) { in apply_impl_helper()
283 double output_scale, in apply_impl()
323 double output_scale, in apply()
331 double output_scale, in apply_relu()
355 double output_scale, in run()
378 double output_scale, in run()
H A DLinear.cpp93 …ply_impl_helper(const at::Tensor& quantized_output, const at::Tensor& input, double output_scale) { in apply_impl_helper()
301 double output_scale, in apply_impl()
326 double output_scale, in apply()
333 double output_scale, in apply_relu()
348 double output_scale, in run()
/aosp_15_r20/external/XNNPACK/src/operators/
H A Dlut-elementwise-nc.c28 float output_scale, in create_lut_elementwise_nc()
151 float output_scale, in xnn_create_elu_nc_qs8()
185 float output_scale, in xnn_create_sigmoid_nc_qs8()
222 float output_scale, in xnn_create_sigmoid_nc_qu8()
263 float output_scale, in xnn_create_tanh_nc_qs8()
300 float output_scale, in xnn_create_tanh_nc_qu8()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/ao_sparse/quantized/cpu/
H A Dqlinear.cpp28 double output_scale, in apply_impl()
217 double output_scale, in apply()
224 double output_scale, in apply_relu()
239 double output_scale, in run()
/aosp_15_r20/external/executorch/backends/cadence/reference/operators/
H A Dquantized_layer_norm.cpp34 double output_scale, in quantized_layer_norm_per_tensor_()
95 double output_scale, in quantized_layer_norm_()
124 double output_scale, in quantized_layer_norm_out()
166 double output_scale, in quantized_layer_norm_per_tensor_out()
/aosp_15_r20/external/XNNPACK/test/
H A Dsigmoid.cc30 const float output_scale = 0x1.0p-8f; in TEST_F() local
70 const float output_scale = 0x1.0p-8f; in TEST_F() local
148 const float output_scale = 0x1.0p-8f; in TEST_F() local
211 const float output_scale = 0x1.0p-8f; in TEST_F() local
H A Dleaky-relu.cc71 const float output_scale = scale_dist(rng); in TEST_F() local
112 const float output_scale = scale_dist(rng); in TEST_F() local
214 const float output_scale = scale_dist(rng); in TEST_F() local
280 const float output_scale = scale_dist(rng); in TEST_F() local
H A Dclamp.cc31 const float output_scale = input_scale; in TEST_F() local
75 const float output_scale = input_scale; in TEST_F() local
157 const float output_scale = input_scale; in TEST_F() local
222 const float output_scale = input_scale; in TEST_F() local
H A Dstatic-transpose.cc50 const float output_scale = input_scale; in TEST_F() local
98 const float output_scale = input_scale; in TEST_F() local
188 const float output_scale = input_scale; in TEST_F() local
253 const float output_scale = input_scale; in TEST_F() local
H A Dmultiply2.cc32 const float output_scale = scale_dist(rng); in TEST_F() local
86 const float output_scale = scale_dist(rng); in TEST_F() local
187 const float output_scale = scale_dist(rng); in TEST_F() local
267 const float output_scale = scale_dist(rng); in TEST_F() local
H A Du32-vlog.cc44 for (uint32_t output_scale = 0; output_scale < 65536; output_scale += 3) { in TEST() local
102 for (uint32_t output_scale = 0; output_scale < 65536; output_scale += 5) { in TEST() local
160 for (uint32_t output_scale = 0; output_scale < 65536; output_scale += 5) { in TEST() local
218 for (uint32_t output_scale = 0; output_scale < 65536; output_scale += 7) { in TEST() local
H A Dsubtract2.cc33 const float output_scale = scale_dist(rng); in TEST_F() local
89 const float output_scale = scale_dist(rng); in TEST_F() local
192 const float output_scale = scale_dist(rng); in TEST_F() local
272 const float output_scale = scale_dist(rng); in TEST_F() local
H A Ddepth-to-space.cc102 const float output_scale = input_scale; in TEST_F() local
142 const float output_scale = input_scale; in TEST_F() local
220 const float output_scale = input_scale; in TEST_F() local
284 const float output_scale = input_scale; in TEST_F() local
H A Dvlog-microkernel-tester.h48 inline VLogMicrokernelTester& output_scale(uint32_t output_scale) { in output_scale() argument
53 inline uint32_t output_scale() const { in output_scale() function
H A Dconvert.cc143 const float output_scale = std::uniform_real_distribution<float>(0.0883883f, 11.3137f)(rng); in TEST_F() local
222 const float output_scale = std::uniform_real_distribution<float>(0.0883883f, 11.3137f)(rng); in TEST_F() local
438 const float output_scale = std::uniform_real_distribution<float>(0.25f, 4.0f)(rng); in TEST_F() local
564 const float output_scale = std::uniform_real_distribution<float>(0.25f, 4.0f)(rng); in TEST_F() local
H A Delu-operator-tester.h106 inline ELUOperatorTester& output_scale(float output_scale) { in output_scale() argument
113 inline float output_scale() const { in output_scale() function
H A Dleaky-relu-operator-tester.h113 inline LeakyReLUOperatorTester& output_scale(float output_scale) { in output_scale() argument
120 inline float output_scale() const { in output_scale() function
/aosp_15_r20/external/executorch/backends/cadence/hifi/operators/
H A Dquantized_layer_norm.cpp37 double output_scale, in quantized_layer_norm_per_tensor_()
102 double output_scale, in quantized_layer_norm_()
131 double output_scale, in quantized_layer_norm_out()
169 double output_scale, in quantized_layer_norm_per_tensor_out()
/aosp_15_r20/external/pytorch/torch/ao/nn/intrinsic/quantized/modules/
H A Dlinear_relu.py51 def from_reference(cls, ref_linear_relu, output_scale, output_zero_point): argument
117 def from_reference(cls, ref_mod, output_scale, output_zero_point): argument
180 def from_reference(cls, ref_mod, output_scale, output_zero_point): argument

1234567