Searched defs:dims_to_sum (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/ |
| H A D | SparseCsrTensorMath.cu | 673 Tensor reduce_sparse_csr_cuda_template(const Tensor& sparse, IntArrayRef dims_to_sum, bool keepdim,… in reduce_sparse_csr_cuda_template()
775 Tensor _sparse_csr_sum_cuda(const Tensor& input, IntArrayRef dims_to_sum, bool keepdim, std::option… in _sparse_csr_sum_cuda()
|
| H A D | SparseCUDATensorMath.cu | 540 …parse_sum_backward_cuda(const Tensor& grad_, const SparseTensor& input_, IntArrayRef dims_to_sum) { in _sparse_sum_backward_cuda()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/ |
| H A D | SparseCsrTensorMath.cpp | 1296 Tensor reduce_sparse_csr_cpu_template(const Tensor& sparse, IntArrayRef dims_to_sum, bool keepdim, … in reduce_sparse_csr_cpu_template()
1331 Tensor _sparse_csr_sum_cpu(const Tensor& input, IntArrayRef dims_to_sum, bool keepdim, std::optiona… in _sparse_csr_sum_cpu()
|
| H A D | SparseTensorMath.cpp | 1640 Tensor _sparse_sum(const SparseTensor& input, IntArrayRef dims_to_sum, ScalarType dtype) { in _sparse_sum()
1644 Tensor _sparse_sum(const SparseTensor& input, IntArrayRef dims_to_sum) { in _sparse_sum()
1759 …sparse_sum_backward_cpu(const Tensor& grad_, const SparseTensor& input_, IntArrayRef dims_to_sum) { in _sparse_sum_backward_cpu()
|