Searched defs:batchPrefix (Results 1 – 2 of 2) sorted by relevance
| /llvm-project/mlir/lib/Dialect/SparseTensor/Transforms/Utils/ |
| H A D | SparseTensorIterator.cpp | 63 Value peekCrdAt(OpBuilder &b, Location l, ValueRange batchPrefix, in peekCrdAt()
97 ValuePair peekRangeAt(OpBuilder &b, Location l, ValueRange batchPrefix, in peekRangeAt()
133 ValuePair peekRangeAt(OpBuilder &b, Location l, ValueRange batchPrefix, in peekRangeAt()
179 ValuePair peekRangeAt(OpBuilder &b, Location l, ValueRange batchPrefix, in peekRangeAt()
201 ValuePair peekRangeAt(OpBuilder &b, Location l, ValueRange batchPrefix, in peekRangeAt()
215 collapseRangeBetween(OpBuilder &b, Location l, ValueRange batchPrefix, in collapseRangeBetween()
228 ValuePair peekRangeAt(OpBuilder &b, Location l, ValueRange batchPrefix, in peekRangeAt()
461 ValueRange batchPrefix = parent ? parent->getBatchCrds() : ValueRange{}; in genInitImpl() local
1315 ValueRange batchPrefix = parent ? parent->getBatchCrds() : ValueRange{}; in genInitImpl() local
|
| H A D | SparseTensorIterator.h | 57 collapseRangeBetween(OpBuilder &b, Location l, ValueRange batchPrefix, in collapseRangeBetween() argument
|