Searched defs:SparseTensorImpl (Results 1 – 2 of 2) sorted by relevance
17 struct TORCH_API SparseTensorImpl : public TensorImpl { struct49 explicit SparseTensorImpl(at::DispatchKeySet, const caffe2::TypeMeta); argument53 int64_t nnz() const { in nnz()57 c10::SymInt sym_nnz() const { in sym_nnz()60 int64_t sparse_dim() const { in sparse_dim()63 int64_t dense_dim() const { in dense_dim()66 bool coalesced() const { in coalesced()69 Tensor indices() const { in indices()72 Tensor values() const { in values()86 void raw_resize_(int64_t sparse_dim, int64_t dense_dim, IntArrayRef size) { in raw_resize_()[all …]
30 SparseTensorImpl::SparseTensorImpl(at::DispatchKeySet key_set, const caffe2::TypeMeta data_type) in SparseTensorImpl() function in at::SparseTensorImpl35 SparseTensorImpl::SparseTensorImpl(at::DispatchKeySet key_set, const caffe2::TypeMeta data_type, at… in SparseTensorImpl() function in at::SparseTensorImpl