Lines Matching refs:enumerator
529 SparseTensorEnumeratorBase<V> *enumerator; in toCOO() local
530 newEnumerator(&enumerator, getRank(), perm); in toCOO()
532 new SparseTensorCOO<V>(enumerator->permutedSizes(), values.size()); in toCOO()
533 enumerator->forallElements([&coo](const std::vector<uint64_t> &ind, V val) { in toCOO()
540 delete enumerator; in toCOO()
583 SparseTensorEnumeratorBase<V> *enumerator; in newSparseTensor() local
584 source->newEnumerator(&enumerator, rank, perm); in newSparseTensor()
585 const auto &permsz = enumerator->permutedSizes(); in newSparseTensor()
589 delete enumerator; in newSparseTensor()
953 void initialize(SparseTensorEnumeratorBase<V> &enumerator) { in initialize() argument
954 assert(enumerator.getRank() == getRank() && "Tensor rank mismatch"); in initialize()
955 assert(enumerator.permutedSizes() == dimSizes && "Tensor size mismatch"); in initialize()
956 enumerator.forallElements( in initialize()
1015 SparseTensorEnumeratorBase<V> *enumerator; in SparseTensorStorage() local
1016 tensor.newEnumerator(&enumerator, getRank(), perm); in SparseTensorStorage()
1020 nnz.initialize(*enumerator); in SparseTensorStorage()
1052 enumerator->forallElements([this](const std::vector<uint64_t> &ind, V val) { in SparseTensorStorage()
1078 delete enumerator; in SparseTensorStorage()