Searched refs:inductionVariables (Results 1 – 2 of 2) sorted by relevance
265 llvm::SmallVector<mlir::Value> inductionVariables; in loopOverElementsBase() local284 inductionVariables.push_back(loop.getInductionVar()); in loopOverElementsBase()286 assert(outerLoop && !inductionVariables.empty() && in loopOverElementsBase()290 getElementAt(builder, loc, shape, slice, inductionVariables); in loopOverElementsBase()386 mlir::Value slice, mlir::ValueRange inductionVariables) { in getElementAt() argument390 size_t inductionIdx = inductionVariables.size() - 1; in getElementAt()394 indexes.emplace_back(inductionVariables[inductionIdx--]); in getElementAt()397 mlir::Value vecIndex = inductionVariables[inductionIdx--]; in getElementAt()
121 mlir::ValueRange inductionVariables);