Lines Matching defs:OpBuilder

100 static AffineMap unpackedPermutationMap(OpBuilder &b, OpTy xferOp) {
115 static void getXferIndices(OpBuilder &b, OpTy xferOp, Value iv,
134 static void maybeYieldValue(OpBuilder &b, Location loc, bool hasRetVal,
151 static Value generateMaskCheck(OpBuilder &b, OpTy xferOp, Value iv) {
189     OpBuilder &b, OpTy xferOp, Value iv, std::optional<int64_t> dim,
191     function_ref<Value(OpBuilder &, Location)> inBoundsCase,
192     function_ref<Value(OpBuilder &, Location)> outOfBoundsCase = nullptr) {
225         [&](OpBuilder &b, Location loc) {
229         [&](OpBuilder &b, Location loc) {
248     OpBuilder &b, OpTy xferOp, Value iv, std::optional<int64_t> dim,
249     function_ref<void(OpBuilder &, Location)> inBoundsCase,
250     function_ref<void(OpBuilder &, Location)> outOfBoundsCase = nullptr) {
254       [&](OpBuilder &b, Location loc) {
259       [&](OpBuilder &b, Location loc) {
267 static ArrayAttr dropFirstElem(OpBuilder &b, ArrayAttr attr) {
276 static void maybeApplyPassLabel(OpBuilder &b, OpTy newXferOp,
300 static BufferAllocs allocBuffers(OpBuilder &b, OpTy xferOp) {
302   OpBuilder::InsertionGuard guard(b);
413   static TransferReadOp rewriteOp(OpBuilder &b,
441   static Value handleOutOfBoundsDim(OpBuilder &b, TransferReadOp xferOp,
501   static TransferWriteOp rewriteOp(OpBuilder &b,
528   static Value handleOutOfBoundsDim(OpBuilder &b, TransferWriteOp xferOp,
798                                  [&](OpBuilder &builder, Location loc) {
952         [&](OpBuilder &b, Location loc, Value iv, ValueRange loopState) {
959               [&](OpBuilder &b, Location loc) {
970                   OpBuilder::InsertionGuard guard(b);
986               [&](OpBuilder &b, Location /*loc*/) {
1144         [&](OpBuilder &b, Location loc, Value iv, ValueRange loopIterArgs) {
1189 static void maybeAssignMask(OpBuilder &b, OpTy xferOp, OpTy newXferOp,
1203     OpBuilder::InsertionGuard guard(b);
1325           [&](OpBuilder &b, Location loc) {
1345           [&](OpBuilder &b, Location loc) {
1458           [&](OpBuilder &b, Location loc) {
1491           [&](OpBuilder &b, Location loc) {
1518 get1dMemrefIndices(OpBuilder &b, OpTy xferOp, Value iv,
1551   static void generateForLoopBody(OpBuilder &b, Location loc,
1563         [&](OpBuilder &b, Location loc) {
1569         [&](OpBuilder & /*b*/, Location loc) { return vec; });
1573   static Value initialLoopState(OpBuilder &b, TransferReadOp xferOp) {
1584   static void generateForLoopBody(OpBuilder &b, Location loc,
1593         /*inBoundsCase=*/[&](OpBuilder &b, Location loc) {
1601   static Value initialLoopState(OpBuilder &b, TransferWriteOp xferOp) {
1673         [&](OpBuilder &b, Location loc, Value iv, ValueRange loopState) {