Home
last modified time | relevance | path

Searched refs:SparseCsrTensor (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/
H A DSparseBlasImpl.h11 const at::sparse_csr::SparseCsrTensor& mat1,
18 const at::sparse_csr::SparseCsrTensor& mat,
25 const at::sparse_csr::SparseCsrTensor& mat1,
26 const at::sparse_csr::SparseCsrTensor& mat2,
29 const at::sparse_csr::SparseCsrTensor& result);
32 const at::sparse_csr::SparseCsrTensor& A,
44 const at::sparse_csr::SparseCsrTensor& result);
H A DSparseBlasImpl.cpp73 const at::sparse_csr::SparseCsrTensor& mat1, in addmm_out_legacy()
130 const at::sparse_csr::SparseCsrTensor& A, in block_sparse_triangular_solve_vec()
258 const at::sparse_csr::SparseCsrTensor& A, in block_sparse_triangular_solve_mat()
399 const at::sparse_csr::SparseCsrTensor& mat, in block_sparse_mv()
464 const at::sparse_csr::SparseCsrTensor& mat1, in block_sparse_mm()
580 const at::sparse_csr::SparseCsrTensor& mat1, in spmm()
690 const at::sparse_csr::SparseCsrTensor& A, in spgemm()
691 const at::sparse_csr::SparseCsrTensor& B, in spgemm()
694 const at::sparse_csr::SparseCsrTensor& C) { in spgemm()
975 const at::sparse_csr::SparseCsrTensor& mat, in addmv_out_sparse_csr()
[all …]
H A DSparseCsrTensorMath.cu142 const SparseCsrTensor& src, in add_out_dense_sparse_compressed_cuda()
312 const SparseCsrTensor& other, in add_out_sparse_compressed_cuda()
314 SparseCsrTensor& out) { in add_out_sparse_compressed_cuda()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/
H A DSparseCsrTensor.cpp349 static SparseCsrTensor new_compressed_tensor(const TensorOptions& options) { in new_compressed_tensor()
443 SparseCsrTensor self = new_compressed_tensor(options); in sparse_compressed_tensor_with_dims()
470 SparseCsrTensor self = new_compressed_tensor(options); in _sparse_compressed_tensor_unsafe_symint()
494 SparseCsrTensor self = new_compressed_tensor(options); in _sparse_compressed_tensor_unsafe_template()
804 int64_t _nnz_sparse_csr(const SparseCsrTensor& self) { in _nnz_sparse_csr()
852 int64_t sparse_dim_sparse_csr(const SparseCsrTensor& self) { in sparse_dim_sparse_csr()
856 int64_t dense_dim_sparse_csr(const SparseCsrTensor& self) { in dense_dim_sparse_csr()
860 const SparseCsrTensor& resize_as_sparse_compressed_( in resize_as_sparse_compressed_()
861 const SparseCsrTensor& self, in resize_as_sparse_compressed_()
862 const SparseCsrTensor& src) { in resize_as_sparse_compressed_()
[all …]
H A DSparseCsrTensorMath.cpp723 const SparseCsrTensor& sparse, in addmm_sparse_compressed_dense()
814 const SparseCsrTensor& src, in add_out_dense_sparse_compressed_cpu()
954 const SparseCsrTensor& other, in add_out_sparse_compressed_cpu()
956 SparseCsrTensor& out) { in add_out_sparse_compressed_cpu()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mkl/
H A DSparseCsrLinearAlgebra.cpp16 const SparseCsrTensor& sparse_, in _sparse_mm_mkl_()
219 const SparseCsrTensor& sparse_, in _sparse_mm_mkl_()
H A DSparseCsrLinearAlgebra.h9 const SparseCsrTensor& sparse_,
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DSparseCsrTensorUtils.h157 using SparseCsrTensor = Tensor; variable
175 inline SparseCsrTensorImpl* get_sparse_csr_impl(const SparseCsrTensor& self) { in get_sparse_csr_impl()
/aosp_15_r20/external/pytorch/
H A Dbuild_variables.bzl1408 "aten/src/ATen/native/sparse/SparseCsrTensor.cpp",