Searched defs:SparseTensorCOO (Results 1 – 1 of 1) sorted by relevance
141 struct SparseTensorCOO final { struct143 SparseTensorCOO(const std::vector<uint64_t> &dimSizes, uint64_t capacity) in SparseTensorCOO() argument152 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()204 void startIterator() { in startIterator()210 const Element<V> *getNext() { in getNext()238 std::vector<Element<V>> elements; // all COO elements239 std::vector<uint64_t> indices; // shared index pool240 bool iteratorLocked = false;[all …]