Lines Matching defs:initArgs
1933 ValueRange initArgs, AffineMapAttr order,
1936 build(builder, result, initArgs.getTypes(), tensor, initArgs, order);
1948 blockArgTypes.append(initArgs.getTypes().begin(), initArgs.getTypes().end());
2225 SmallVector<OpAsmParser::UnresolvedOperand> initArgs;
2245 if (parser.parseAssignmentList(blockArgs, initArgs))
2280 if (args.size() != initArgs.size() || args.size() != state.types.size()) {
2286 for (auto [it, init, tp] : llvm::zip_equal(args, initArgs, state.types)) {
2311 SmallVector<OpAsmParser::UnresolvedOperand> initArgs;
2314 if (parser.parseAssignmentList(blockArgs, initArgs))
2342 if (args.size() != initArgs.size() || args.size() != state.types.size()) {
2348 for (auto [it, init, tp] : llvm::zip_equal(args, initArgs, state.types)) {
2440 Value iterSpace, ValueRange initArgs) {
2444 return build(builder, odsState, iterSpace, initArgs, set);
2448 Value iterSpace, ValueRange initArgs,
2453 odsState.addOperands(initArgs);
2457 odsState.addTypes(initArgs.getTypes());
2461 for (Value v : initArgs)
2559 auto initArgs = getInitArgs();
2563 if (!llvm::all_equal({initArgs.size(), iterArgs.size(), yieldVals.size(),
2569 llvm::enumerate(initArgs, iterArgs, yieldVals, opResults)) {
2617 ValueRange iterSpaces, ValueRange initArgs,
2629 return CoIterateOp::build(builder, odsState, initArgs.getTypes(), iterSpaces,
2630 initArgs, set, cases,
2721 auto initArgs = getInitArgs();
2725 if (!llvm::all_equal({initArgs.size(), iterArgs.size(), yieldVals.size(),
2733 llvm::enumerate(initArgs, iterArgs, yieldVals, opResults)) {