Searched defs:dimSizes (Results 1 – 2 of 2) sorted by relevance
103 assertPermutedSizesMatchShape(const std::vector<uint64_t> &dimSizes, in assertPermutedSizesMatchShape()237 const std::vector<uint64_t> dimSizes; // per-dimension sizes member266 SparseTensorStorageBase(const std::vector<uint64_t> &dimSizes, in SparseTensorStorageBase()369 const std::vector<uint64_t> dimSizes; member in __anon2f09016c0111::SparseTensorStorageBase394 SparseTensorStorage(const std::vector<uint64_t> &dimSizes, in SparseTensorStorage()405 SparseTensorStorage(const std::vector<uint64_t> &dimSizes, in SparseTensorStorage()912 SparseTensorNNZ(const std::vector<uint64_t> &dimSizes, in SparseTensorNNZ()1005 const std::vector<uint64_t> &dimSizes; member in __anon2f09016c0111::SparseTensorNNZ1012 const std::vector<uint64_t> &dimSizes, const uint64_t *perm, in SparseTensorStorage()1403 auto &dimSizes = coo->getDimSizes(); in outSparseTensor() local[all …]
725 auto dimSizes = op.getMaskDimSizes(); in matchAndRewrite() local