Searched defs:w_scale (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | RowwiseScaledMM.cu | 142 at::Tensor w_scale, in f8f8bf16_rowwise_impl()
310 at::Tensor w_scale, in dispatch_fp8_rowwise_kernel_on_tile_size()
347 at::Tensor w_scale, in handle_transposition()
374 at::Tensor w_scale, in dispatch_fp8_rowwise_kernel_on_cluster_size_and_transpose()
459 at::Tensor w_scale, in dispatch_fp8_rowwise_kernel_on_fast_accum()
479 at::Tensor w_scale, in dispatch_fp8_rowwise_kernel_on_input_dtypes()
500 at::Tensor w_scale, in dispatch_fp8_rowwise_kernel_on_bias_dtype()
573 at::Tensor w_scale, // FP32 in f8f8bf16_rowwise()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | fbgemm_utils.h | 42 std::vector<float> w_scale; member
205 std::vector<float> w_scale; member
378 std::vector<float> w_scale; member
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ao_sparse/quantized/cpu/ |
| H A D | fbgemm_utils.h | 36 std::vector<float> w_scale; member
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/test/ |
| H A D | vulkan_quantized_api_test.cpp | 1508 const double w_scale = 0.1; in test_conv2d() local
1616 const double w_scale = 0.1; in TEST_F() local
1719 const double w_scale = 0.1; in TEST_F() local
2491 double w_scale = 0.29, in test_quantized_conv2d()
3327 double w_scale = std::get<0>(weight_quant_params); in _test_quantized_linear() local
|
| /aosp_15_r20/external/ComputeLibrary/examples/ |
| H A D | graph_ssd_mobilenet.cpp | 676 const float w_scale = 5.f; in create_graph_qasymm() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| H A D | image_resize_ops.cc | 561 float w_scale; in GeneralCompile() local
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | mkldnn_ir.py | 1010 w_scale, argument
|
| H A D | ir.py | 6935 w_scale, argument
|
| /aosp_15_r20/external/pytorch/torch/ |
| H A D | _meta_registrations.py | 2330 w_scale, argument
2366 w_scale, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/ |
| H A D | QuantizedOpKernels.cpp | 671 float w_scale = static_cast<float>(qw.q_scale()); in qprelu_out_kernel() local
|