Home
last modified time | relevance | path

Searched defs:scales_and_zeros (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/executorch/backends/vulkan/test/op_tests/
H A Dlinear_weight_int4_test.cpp27 const at::Tensor& scales_and_zeros, in linear_weight_int4_reference_impl()
47 const at::Tensor& scales_and_zeros, in dequantize_and_linear()
96 at::Tensor scales_and_zeros = in test_reference_linear_int4() local
142 at::Tensor scales_and_zeros = in test_vulkan_linear_int4() local
/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/
H A DQuantizedLinear.cpp238 const ValueRef scales_and_zeros, in check_q_4w_linear_args()
306 ValueRef scales_and_zeros = prepack_standard( in add_q_4w_linear_node() local
/aosp_15_r20/external/executorch/backends/vulkan/runtime/graph/ops/impl/utils/
H A DQPackUtils.cpp50 const float* scales_and_zeros) { in int4mm_dequantize_weights()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A Dint4mm_kernel.cpp406 … auto scales_and_zeros = load_as_float32x4x2(ScaleAndZeros + kb * ldc * 2 + n * 2 + i * 8); in tinygemm_kernel_() local