Lines Matching defs:TensorLevel
26 using TensorLevel = unsigned;
126 ArrayRef<TensorLevel> tidLvls);
134 TensorLevel tidLvl, AffineExpr lvlExpr);
147 OpBuilder &builder, Location loc, ArrayRef<TensorLevel> tidLvls,
202 /// Compresses a TensorId and Level into a TensorLevel.
203 TensorLevel makeTensorLevel(TensorId t, Level l) const {
207 /// De-compresses a TensorLevel back to a pair of TensorId and Level.
208 std::pair<TensorId, Level> unpackTensorLevel(TensorLevel tidLvl) const {
213 /// Converts a range of TensorLevel to a range of std::pair<TensorId, Level>
217 static_assert(std::is_same_v<llvm::remove_cvref_t<EltTy>, TensorLevel>,
218 "Must be unpacking a TensorLevel range");
255 LoopInfo(ArrayRef<TensorLevel> tidLvls, Operation *loop, Block *userBlock,
266 const llvm::SmallVector<TensorLevel> tidLvls;
272 void categorizeIterators(ArrayRef<TensorLevel> tidLvls,
429 std::vector<std::pair<Value, std::vector<TensorLevel>>> loopSeqStack;