Searched refs:_mul_sparse_sparse_zero_dim_out (Results 1 – 3 of 3) sorted by relevance
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
H A D | SparseTensorMath.h | 10 TORCH_API sparse::SparseTensor& _mul_sparse_sparse_zero_dim_out(const Tensor& zero_dim, const Tenso…
|
H A D | SparseTensorMath.cpp | 1002 Tensor& _mul_sparse_sparse_zero_dim_out(const Tensor& zero_dim, const Tensor& other, Tensor& r) { in _mul_sparse_sparse_zero_dim_out() function 1064 return _mul_sparse_sparse_zero_dim_out(src_, t_, r); in mul_out_sparse_cpu() 1067 return _mul_sparse_sparse_zero_dim_out(t_, src_, r); in mul_out_sparse_cpu()
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/ |
H A D | SparseCUDATensorMath.cu | 476 return _mul_sparse_sparse_zero_dim_out(src_, t_, r_); in mul_out_sparse_cuda() 479 return _mul_sparse_sparse_zero_dim_out(t_, src_, r_); in mul_out_sparse_cuda()
|