Lines Matching defs:inArgs

203     llvm::SmallVector<IntrinsicArgument, 3> inArgs;
204     inArgs.push_back({operation.getArray(), operation.getArray().getType()});
205     inArgs.push_back({operation.getDim(), i32});
206     inArgs.push_back({operation.getMask(), logicalType});
208     return lowerArguments(operation, inArgs, rewriter, argLowering);
214     llvm::SmallVector<IntrinsicArgument, 3> inArgs;
215     inArgs.push_back({operation.getArray(), operation.getArray().getType()});
216     inArgs.push_back({operation.getDim(), i32});
217     inArgs.push_back({operation.getMask(), logicalType});
220     inArgs.push_back({kind, i32});
221     inArgs.push_back({operation.getBack(), i32});
223     return lowerArguments(operation, inArgs, rewriter, argLowering);
229     llvm::SmallVector<IntrinsicArgument, 2> inArgs;
230     inArgs.push_back({operation.getMask(), logicalType});
231     inArgs.push_back({operation.getDim(), i32});
233     return lowerArguments(operation, inArgs, rewriter, argLowering);
323     llvm::SmallVector<IntrinsicArgument, 3> inArgs;
324     inArgs.push_back({count.getMask(), logicalType});
325     inArgs.push_back({count.getDim(), i32});
328     inArgs.push_back({kind, i32});
332         lowerArguments(count, inArgs, rewriter, argLowering);
355     llvm::SmallVector<IntrinsicArgument, 2> inArgs;
356     inArgs.push_back({lhs, lhs.getType()});
357     inArgs.push_back({rhs, rhs.getType()});
361         lowerArguments(matmul, inArgs, rewriter, argLowering);
386     llvm::SmallVector<IntrinsicArgument, 2> inArgs;
387     inArgs.push_back({lhs, lhs.getType()});
388     inArgs.push_back({rhs, rhs.getType()});
392         lowerArguments(dotProduct, inArgs, rewriter, argLowering);
416     llvm::SmallVector<IntrinsicArgument, 1> inArgs;
417     inArgs.push_back({arg, arg.getType()});
421         lowerArguments(transpose, inArgs, rewriter, argLowering);
447     llvm::SmallVector<IntrinsicArgument, 2> inArgs;
448     inArgs.push_back({lhs, lhs.getType()});
449     inArgs.push_back({rhs, rhs.getType()});
453         lowerArguments(multranspose, inArgs, rewriter, argLowering);
475     llvm::SmallVector<IntrinsicArgument, 3> inArgs;
477     inArgs.push_back({array, array.getType()});
479     inArgs.push_back({shift, shift.getType()});
480     inArgs.push_back({cshift.getDim(), builder.getI32Type()});
484         lowerArguments(cshift, inArgs, rewriter, argLowering);
506     llvm::SmallVector<IntrinsicArgument, 4> inArgs;
508     inArgs.push_back({array, array.getType()});
510     inArgs.push_back({shape, shape.getType()});
513     inArgs.push_back({pad, pad ? pad.getType() : noneType});
515     inArgs.push_back({order, order ? order.getType() : noneType});
519         lowerArguments(reshape, inArgs, rewriter, argLowering);