Searched defs:scale_b (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | Blas.cpp | 905 const at::Tensor& scale_b, in get_scaling_type()
989 const Tensor& scale_b, in _scaled_mm_out_cuda()
1192 const Tensor& scale_b, in _scaled_mm_cuda()
|
| H A D | RowwiseScaledMM.cu | 517 const at::Tensor& scale_b, in check_inputs()
|
| /aosp_15_r20/external/pytorch/torch/csrc/inductor/aoti_torch/ |
| H A D | shim_common.cpp | 689 AtenTensorHandle scale_b, in aoti_torch__scaled_mm()
719 AtenTensorHandle scale_b, 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
|