Lines Matching refs:SparseTensorStorage
387 class SparseTensorStorage final : public SparseTensorStorageBase { class
394 SparseTensorStorage(const std::vector<uint64_t> &dimSizes, in SparseTensorStorage() function in __anon2f09016c0111::SparseTensorStorage
405 SparseTensorStorage(const std::vector<uint64_t> &dimSizes, in SparseTensorStorage() function in __anon2f09016c0111::SparseTensorStorage
408 : SparseTensorStorage(dimSizes, perm, sparsity) { in SparseTensorStorage()
451 SparseTensorStorage(const std::vector<uint64_t> &dimSizes,
455 ~SparseTensorStorage() final = default;
551 static SparseTensorStorage<P, I, V> *
554 SparseTensorStorage<P, I, V> *n = nullptr; in newSparseTensor()
558 n = new SparseTensorStorage<P, I, V>(coosz, perm, sparsity, coo); in newSparseTensor()
566 n = new SparseTensorStorage<P, I, V>(permsz, perm, sparsity, coo); in newSparseTensor()
578 static SparseTensorStorage<P, I, V> *
588 new SparseTensorStorage<P, I, V>(permsz, perm, sparsity, *source); in newSparseTensor()
847 SparseTensorEnumerator(const SparseTensorStorage<P, I, V> &tensor, in SparseTensorEnumerator()
863 static_cast<const SparseTensorStorage<P, I, V> &>(this->src); in forallElements()
1011 SparseTensorStorage<P, I, V>::SparseTensorStorage( in SparseTensorStorage() function in __anon2f09016c0111::SparseTensorStorage
1014 : SparseTensorStorage(dimSizes, perm, sparsity) { in SparseTensorStorage()
1427 static SparseTensorStorage<uint64_t, uint64_t, V> *
1456 auto *tensor = SparseTensorStorage<uint64_t, uint64_t, V>::newSparseTensor( in toMLIRSparseTensor()
1469 static_cast<SparseTensorStorage<uint64_t, uint64_t, V> *>(tensor); in fromMLIRSparseTensor()
1523 auto *tensor = SparseTensorStorage<P, I, V>::newSparseTensor( \
1531 return SparseTensorStorage<P, I, V>::newSparseTensor(rank, shape, perm, \
1536 coo = static_cast<SparseTensorStorage<P, I, V> *>(ptr)->toCOO(perm); \