Lines Matching defs:SparseCsrTensorImpl
27 struct TORCH_API SparseCsrTensorImpl : public TensorImpl { struct
34 explicit SparseCsrTensorImpl( argument
56 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()
95 Layout layout_impl() const override { in layout_impl()
98 void set_layout(Layout layout) { in set_layout()
128 auto impl = c10::make_intrusive<SparseCsrTensorImpl>( in shallow_copy_and_detach_core() argument
171 explicit SparseCsrTensorImpl( argument
189 const SparseCsrTensorImpl* src_sparse_impl, in copy_tensor_metadata() argument