Searched defs:SparseCsrTensorImpl (Results 1 – 2 of 2) sorted by relevance
27 struct TORCH_API SparseCsrTensorImpl : public TensorImpl { struct34 explicit SparseCsrTensorImpl( argument56 const Tensor& compressed_indices() const { in compressed_indices()59 const Tensor& plain_indices() const { in plain_indices()62 const Tensor& values() const { in values()65 int64_t nnz() { in nnz()69 inline int64_t batch_dim() const noexcept { in batch_dim()73 inline int64_t sparse_dim() const noexcept { in sparse_dim()77 inline int64_t dense_dim() const noexcept { in dense_dim()82 inline int64_t block_dim() const noexcept { in block_dim()[all …]
11 SparseCsrTensorImpl::SparseCsrTensorImpl( in SparseCsrTensorImpl() function in at::SparseCsrTensorImpl40 SparseCsrTensorImpl::SparseCsrTensorImpl( in SparseCsrTensorImpl() function in at::SparseCsrTensorImpl