Searched refs:inductionVariables (Results 1 – 2 of 2) sorted by relevance
267 llvm::SmallVector<mlir::Value> inductionVariables; in loopOverElementsBase() local286 inductionVariables.push_back(loop.getInductionVar()); in loopOverElementsBase()288 assert(outerLoop && !inductionVariables.empty() && in loopOverElementsBase()292 getElementAt(builder, loc, shape, slice, inductionVariables); in loopOverElementsBase()388 mlir::Value slice, mlir::ValueRange inductionVariables) { in getElementAt() argument392 size_t inductionIdx = inductionVariables.size() - 1; in getElementAt()397 indexes.emplace_back(inductionVariables[inductionIdx--]); in getElementAt()400 mlir::Value vecIndex = inductionVariables[inductionIdx--]; in getElementAt()
121 mlir::ValueRange inductionVariables);