Searched defs:scale_a (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Blas.cpp | 904 const at::Tensor& scale_a, in get_scaling_type()
988 const Tensor& scale_a, in _scaled_mm_out_cuda()
1191 const Tensor& scale_a, in _scaled_mm_cuda()
|
| H A D | RowwiseScaledMM.cu | 516 const at::Tensor& scale_a, in check_inputs()
|
| /aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_torch/ |
| H A D | shim_common.cpp | 688 AtenTensorHandle scale_a, in aoti_torch__scaled_mm()
718 AtenTensorHandle scale_a, in aoti_torch__scaled_mm_v2()
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_aot_inductor.py | 720 def forward(self, x, weight, bias, scale_a, scale_b): argument
768 def forward(self, x, bias, scale_a, scale_b): argument
|