| /aosp_15_r20/external/pytorch/torch/ao/quantization/pt2e/representation/ |
| H A D | rewrite.py | 51 weight_scale, argument
86 weight_scale, argument
152 weight_scale, argument
185 weight_scale, argument
252 weight_scale, argument
303 weight_scale, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | qlinear_prepack.cpp | 366 const at::Tensor& weight_scale, in wrapped_quantized_linear()
422 [[maybe_unused]] const at::Tensor& weight_scale, in wrapped_quantized_linear_meta()
445 const at::Tensor& weight_scale, in _wrapped_linear_prepack()
516 [[maybe_unused]] const at::Tensor& weight_scale, in _wrapped_linear_prepack_meta()
|
| H A D | qembeddingbag_prepack.cpp | 129 at::Half weight_scale = weight_scales[row]; in prepack() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | QuantizedLinear.cpp | 55 const Scalar& weight_scale, in fbgemm_linear_int8_weight_fp32_activation()
192 const Scalar& weight_scale, in fbgemm_linear_int8_weight()
|
| /aosp_15_r20/external/mesa3d/src/gallium/drivers/etnaviv/ |
| H A D | etnaviv_ml_nn.c | 432 compute_addition_offset(float input1_scale, float input2_scale, float weight_scale) in compute_addition_offset()
440 compute_weight_add(float input1_scale, float input2_scale, float weight_scale) in compute_weight_add()
447 …d(float input1_scale, float input2_scale, uint8_t input1_zp, uint8_t input2_zp, float weight_scale) in compute_bias_add()
|
| H A D | etnaviv_ml.h | 80 float weight_scale; member
|
| /aosp_15_r20/external/libavc/common/ |
| H A D | ih264_trans_macros.h | 102 #define INV_QUANT(i4_value, quant_scale, weight_scale, u4_qp_div_6, rndfactor, qbits)\ argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| H A D | legalize_utils.cc | 105 double weight_scale = weight_per_tensor_qtype.getScale(); in buildRescaleOpConvOutput() local
137 for (double weight_scale : weight_scale_arr) { in buildRescaleOpConvOutput() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/optimize/ |
| H A D | quantization_utils_test.cc | 755 const float weight_scale = 0.5; in TEST_F() local
797 const float weight_scale = 0.5; in TEST_F() local
|
| /aosp_15_r20/external/pytorch/torch/ao/nn/quantized/reference/modules/ |
| H A D | utils.py | 410 weight_scale, argument
|
| /aosp_15_r20/external/executorch/backends/arm/ |
| H A D | tosa_quant_utils.py | 440 weight_scale, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cudnn/ |
| H A D | Conv.cpp | 69 auto weight_scale = maybe_padded_weight_.q_scale(); in apply_impl_helper() local
|
| H A D | Linear.cpp | 98 auto weight_scale = orig_weight.q_scale(); in apply_impl_helper() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/vulkan/ops/ |
| H A D | Convolution.h | 40 float weight_scale; member
|
| /aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_torch/ |
| H A D | shim_common.cpp | 863 AtenTensorHandle weight_scale, in aoti_torch_cpu__wrapped_linear_prepack()
|
| /aosp_15_r20/external/pytorch/torch/onnx/ |
| H A D | symbolic_helper.py | 1618 g: jit_utils.GraphContext, bias, input_scale, weight_scale, axis=None argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | vulkan_quantized_api_test.cpp | 1800 double weight_scale = std::get<0>(weight_quant_params); in test_quantized_conv_transpose2d() local
|