Lines Matching defs:dimSizes
103 assertPermutedSizesMatchShape(const std::vector<uint64_t> &dimSizes, in assertPermutedSizesMatchShape()
237 const std::vector<uint64_t> dimSizes; // per-dimension sizes member
266 SparseTensorStorageBase(const std::vector<uint64_t> &dimSizes, in SparseTensorStorageBase()
369 const std::vector<uint64_t> dimSizes; member in __anon2f09016c0111::SparseTensorStorageBase
394 SparseTensorStorage(const std::vector<uint64_t> &dimSizes, in SparseTensorStorage()
405 SparseTensorStorage(const std::vector<uint64_t> &dimSizes, in SparseTensorStorage()
799 const auto &dimSizes = src.getDimSizes(); // in source storage-order in SparseTensorEnumeratorBase() local
912 SparseTensorNNZ(const std::vector<uint64_t> &dimSizes, in SparseTensorNNZ()
1005 const std::vector<uint64_t> &dimSizes; member in __anon2f09016c0111::SparseTensorNNZ
1012 const std::vector<uint64_t> &dimSizes, const uint64_t *perm, in SparseTensorStorage()
1403 auto &dimSizes = coo->getDimSizes(); in outSparseTensor() local
1845 const uint64_t *dimSizes = stfile.getDimSizes(); in readSparseTensorShape() local