Lines Matching refs:SparseTensorCOO

141 struct SparseTensorCOO final {  struct
143 SparseTensorCOO(const std::vector<uint64_t> &dimSizes, uint64_t capacity) in SparseTensorCOO() argument
224 static SparseTensorCOO<V> *newSparseTensorCOO(uint64_t rank, in newSparseTensorCOO() argument
233 return new SparseTensorCOO<V>(permsz, capacity); in newSparseTensorCOO()
407 SparseTensorCOO<V> *coo) in SparseTensorStorage()
528 SparseTensorCOO<V> *toCOO(const uint64_t *perm) const { in toCOO()
531 SparseTensorCOO<V> *coo = in toCOO()
532 new SparseTensorCOO<V>(enumerator->permutedSizes(), values.size()); in toCOO()
553 const DimLevelType *sparsity, SparseTensorCOO<V> *coo) { in newSparseTensor()
1322 static inline void readCOOValue(SparseTensorCOO<std::complex<V>> *coo, in readCOOValue()
1342 static void inline readCOOValue(SparseTensorCOO<V> *coo, in readCOOValue()
1356 static SparseTensorCOO<V> *
1376 auto *coo = SparseTensorCOO<V>::newSparseTensorCOO(rank, stfile.getDimSizes(), in openSparseTensorCOO()
1399 auto coo = static_cast<SparseTensorCOO<V> *>(tensor); in outSparseTensor()
1447 auto *coo = SparseTensorCOO<V>::newSparseTensorCOO(rank, shape, perm, nse); in toMLIRSparseTensor()
1473 SparseTensorCOO<V> *coo = sparseTensor->toCOO(perm.data()); in fromMLIRSparseTensor()
1513 SparseTensorCOO<V> *coo = nullptr; \
1519 coo = static_cast<SparseTensorCOO<V> *>(ptr); \
1535 return SparseTensorCOO<V>::newSparseTensorCOO(rank, shape, perm); \
1734 static_cast<SparseTensorCOO<V> *>(coo)->add(indices, *value); \ in FOREVERY_O()
1750 static_cast<SparseTensorCOO<V> *>(coo)->getNext(); \
1824 delete static_cast<SparseTensorCOO<V> *>(coo); \