Lines Matching +defs:expanded +defs:c
98 // Extract lhs slice of size {n, wSizeStep, c} @ [0, sw * w + dw * kw, 0]
120 // Extract rhs slice of size [{c, f} for channeled convolutions and {1} for
518 Ncw, // Corresponds to operation that traverses the input in (n, c, w) order.
519 Nwc // Corresponds to operation that traverses the input in (n, w, c) order.
1318 // c. for elementwise, the result is the vector with the firstMaxRankedShape
1418 // 3.c. Not all ops support 0-d vectors, extract the scalar for now.
1445 // 4c. Register CustomVectorizationHook for extractOp.
2157 auto expanded = affine::expandAffineExpr(
2161 rewriter.replaceOp(op, expanded);
3028 /// Op def: ( n, w, c, kw, f )
3030 /// Layout: {{n, strideW * w + dilationW * kw, c}, {kw, c, f}, {n, w, f}}
3037 /// Op def: ( n, c, w, f, kw )
3039 /// Layout: {{n, c, strideW * w + dilationW * kw}, {f, c, kw}, {n, f, w}}
3046 /// Op def: ( n, w, c, kw )
3048 /// Layout: {{n, strideW * w + dilationW * kw, c}, {kw, c}, {n, w, c}}
3118 /// Op def: ( n, w, c, kw, f )
3120 /// Layout: {{n, strideW * w + dilationW * kw, c}, {kw, c, f}, {n, w, f}}
3151 // kernel{kw, c, f}
3182 // kernel{f, c, kw}
3240 // {n,w,c}, weight: {kw,c,f}, output: {n,w,f}. To reuse the base pattern
3283 // Compute contraction: O{n, w, f} += I{n, sw * w + dw * kw, c} * F{c, f}
3364 // Create a contraction: lhs{n, w, c} * rhs{c, f} -> res{n, w, f}
3369 AffineExpr n, w, f, c;
3370 bindDims(ctx, n, w, f, c);
3375 /*indexingMaps=*/MapList{{n, w, c}, {c, f}, {n, w, f}},
3389 // Create a reduction: lhs{n, w, c} -> res{n, w, c}
3401 /// Op def: ( n, w, c, kw)
3403 /// Layout: {{n, strideW * w + dilationW * kw, c}, {kw, c}, {n, w, c}}
3417 // kernel{kw, c}
3432 // out{n, w, c}
3484 // Read lhs slice of size {n, w * strideW + kw * dilationW, c} @ [0, 0,
3491 // Read rhs slice of size {kw, c} @ [0, 0].
3497 // Read res slice of size {n, w, c} @ [0, 0, 0].
3511 // Extract lhs slice of size {n, wSizeStep, c}
3521 // Extract rhs slice of size {c} @ [kw].
3527 // Extract res slice: {n, wSizeStep, c} @ [0, w, 0].
3547 // Compute contraction: O{n, w, c} += I{n, sw * w + dw * kw, c} * F{c}
3580 // Write back res slice: {n, wSizeStep, c} @ [0, w, 0].
3592 // Write back res slice of size {n, w, c} @ [0, 0, 0].
3601 /// * lhs{n, w, c} * rhs{c} -> res{n, w, c} (flatten = false)
3602 /// * lhs{n, w * c} * rhs{c} -> res{n, w * c} (flatten = true)
3668 /// {{n, strideW * w + dilationW * kw, c}, {kw, c, f}, {n, w, f}}
3670 AffineExpr n, w, f, kw, c;
3671 bindDims(ctx, n, w, f, kw, c);
3677 if (layout({/*lhsIndex*/ {n, strideW * w + dilationW * kw, c},
3678 /*rhsIndex*/ {kw, c, f},
3686 /// {{n, c, strideW * w + dilationW * kw}, {f, c, kw}, {n, f, w}}
3688 AffineExpr n, w, f, kw, c;
3689 bindDims(ctx, n, f, w, c, kw);
3694 if (layout({/*lhsIndex*/ {n, c, strideW * w + dilationW * kw},
3695 /*rhsIndex*/ {f, c, kw},
3703 /// {{n, strideW * w + dilationW * kw, c}, {kw}, {n, w, c}} for pooling
3705 AffineExpr n, w, c, kw;
3706 bindDims(ctx, n, w, c, kw);
3712 if (layout({/*lhsIndex*/ {n, strideW * w + dilationW * kw, c},
3714 /*resIndex*/ {n, w, c}}))
3721 /// {{n, c, strideW * w + dilationW * kw}, {kw}, {n, c, w}} for pooling
3723 AffineExpr n, w, c, kw;
3724 bindDims(ctx, n, c, w, kw);
3729 if (layout({/*lhsIndex*/ {n, c, strideW * w + dilationW * kw},
3731 /*resIndex*/ {n, c, w}}))
3738 /// {{n, strideW * w + dilationW * kw, c}, {kw, c}, {n, w, c}}
3742 AffineExpr n, w, c, kw;
3743 bindDims(ctx, n, w, c, kw);
3749 if (layout({/*lhsIndex*/ {n, strideW * w + dilationW * kw, c},
3750 /*rhsIndex*/ {kw, c},
3751 /*resIndex*/ {n, w, c}}))