Home
last modified time | relevance | path

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 DSparseTensorMath.h10 TORCH_API sparse::SparseTensor& _mul_sparse_sparse_zero_dim_out(const Tensor& zero_dim, const Tenso…
H A DSparseTensorMath.cpp1002 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 DSparseCUDATensorMath.cu476 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()