Lines Matching defs:SparseTensorCOO
141 struct SparseTensorCOO final { struct
143 SparseTensorCOO(const std::vector<uint64_t> &dimSizes, uint64_t capacity) in SparseTensorCOO() function
152 void add(const std::vector<uint64_t> &ind, V val) { in add()
178 void sort() { in sort()
195 uint64_t getRank() const { return dimSizes.size(); } in getRank()
198 const std::vector<uint64_t> &getDimSizes() const { return dimSizes; } in getDimSizes()
201 const std::vector<Element<V>> &getElements() const { return elements; } in getElements()
204 void startIterator() { in startIterator()
210 const Element<V> *getNext() { in getNext()
224 static SparseTensorCOO<V> *newSparseTensorCOO(uint64_t rank, in newSparseTensorCOO()
237 const std::vector<uint64_t> dimSizes; // per-dimension sizes
238 std::vector<Element<V>> elements; // all COO elements
239 std::vector<uint64_t> indices; // shared index pool
240 bool iteratorLocked = false;
241 unsigned iteratorPos = 0;