Lines Matching defs:funcOp

72       fir::FirOpBuilder &builder, mlir::func::FuncOp &funcOp, unsigned rank,
254 /// Generate the reduction loop into \p funcOp.
271 genReductionLoop(fir::FirOpBuilder &builder, mlir::func::FuncOp &funcOp,
278 mlir::Block::BlockArgListType args = funcOp.front().getArguments();
288 mlir::Type resultType = funcOp.getResultTypes()[0];
367 /// with signature provided by \p funcOp. The caller is responsible
369 /// \p funcOp is expected to be empty on entry to this function.
372 mlir::func::FuncOp &funcOp, unsigned rank,
406 builder.setInsertionPointToEnd(funcOp.addEntryBlock());
408 genReductionLoop<fir::DoLoopOp, bool, 0>(builder, funcOp, zero, nopLoopCond,
414 mlir::func::FuncOp &funcOp, unsigned rank,
451 builder.setInsertionPointToEnd(funcOp.addEntryBlock());
453 genReductionLoop<fir::DoLoopOp, bool, 0>(builder, funcOp, init, nopLoopCond,
459 mlir::func::FuncOp &funcOp, unsigned rank,
485 builder.setInsertionPointToEnd(funcOp.addEntryBlock());
487 genReductionLoop<fir::DoLoopOp, bool, 0>(builder, funcOp, zero, nopLoopCond,
493 mlir::func::FuncOp &funcOp, unsigned rank,
517 builder.setInsertionPointToEnd(funcOp.addEntryBlock());
521 builder, funcOp, zero, continueCond, ok, genBodyOp, rank, elementType,
526 mlir::func::FuncOp &funcOp, unsigned rank,
548 builder.setInsertionPointToEnd(funcOp.addEntryBlock());
552 builder, funcOp, one, continueCond, ok, genBodyOp, rank, elementType,
659 mlir::func::FuncOp &funcOp, bool isMax,
679 builder.setInsertionPointToEnd(funcOp.addEntryBlock());
681 mlir::Value mask = funcOp.front().getArgument(2);
860 genMinMaxlocReductionLoop(builder, funcOp.front().getArgument(1), init,
869 loc, builder.getRefType(resultBoxTy), funcOp.front().getArgument(0));
882 loc, outputRefTy, funcOp.front().getArgument(0));
899 /// with signature provided by \p funcOp. The caller is responsible
901 /// \p funcOp is expected to be empty on entry to this function.
906 mlir::func::FuncOp &funcOp,
919 mlir::Type resultElementType = funcOp.getResultTypes()[0];
920 builder.setInsertionPointToEnd(funcOp.addEntryBlock());
929 mlir::Block::BlockArgListType args = funcOp.front().getArguments();
1231 mlir::func::FuncOp &funcOp) {
1232 genRuntimeMinMaxlocBody(builder, funcOp, isMax, rank, maskRank, inputType,
1261 mlir::func::FuncOp &funcOp) {
1262 genBodyFunc(builder, funcOp, rank, elementType);
1334 mlir::func::FuncOp &funcOp) {
1335 genRuntimeDotBody(builder, funcOp, *arg1Type, *arg2Type);