Searched defs:b_self (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/ |
| H A D | SparseCUDATensorMath.cu | 137 c10::MaybeOwned<Tensor> b_self = expand_size(self, {mat1.size(0), mat2.size(1)}, "addmm_out"); in addmm_out_sparse_dense_cuda() local
160 c10::MaybeOwned<Tensor> b_self = expand_size(self, {mat1.size(0), mat2.size(1)}, "addmm_out"); in addmm_sparse_dense_cuda() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | LinearAlgebra.cpp | 1590 auto b_self = expand_size(self, {batch1.size(1), batch2.size(2)}, "addbmm_out"); in addbmm_out() local
1610 auto b_self = expand_size(self, {mat1.sizes()[0], mat2.sizes()[1]}, "addmm_out"); in TORCH_IMPL_FUNC() local
1618 auto b_self = expand_size(self, {mat1.sizes()[0], mat2.sizes()[1]}, "addmm_out"); in TORCH_IMPL_FUNC() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
| H A D | SparseTensorMath.cpp | 1319 c10::MaybeOwned<Tensor> b_self = expand_size(self, {mat1.size(0), mat2.size(1)}, "addmm_out"); in addmm_out_sparse_dense_cpu() local
1342 c10::MaybeOwned<Tensor> b_self = expand_size(self, {mat1.size(0), mat2.size(1)}, "addmm_out"); in addmm_sparse_dense_cpu() local
|