Lines Matching defs:dilationW
83 int64_t kwSize, int strideW, int dilationW,
106 /*offsets=*/ArrayRef<int64_t>{0, w * strideW + kw * dilationW, 0},
3030 /// Layout: {{n, strideW * w + dilationW * kw, c}, {kw, c, f}, {n, w, f}}
3032 /// kw is unrolled, w is unrolled iff dilationW > 1.
3039 /// Layout: {{n, c, strideW * w + dilationW * kw}, {f, c, kw}, {n, f, w}}
3041 /// kw is unrolled, w is unrolled iff dilationW > 1.
3048 /// Layout: {{n, strideW * w + dilationW * kw, c}, {kw, c}, {n, w, c}}
3050 /// kw is unrolled, w is unrolled iff dilationW > 1.
3054 int dilationW)
3056 strideW(strideW), dilationW(dilationW) {
3120 /// Layout: {{n, strideW * w + dilationW * kw, c}, {kw, c, f}, {n, w, f}}
3123 /// TODO: w (resp. kw) is unrolled when the strideW ( resp. dilationW) is
3164 ((wSize - 1) * strideW + 1) + ((kwSize - 1) * dilationW + 1) -
3195 ((wSize - 1) * strideW + 1) + ((kwSize - 1) * dilationW + 1) -
3271 kwSize, strideW, dilationW, wSizeStep,
3403 /// Layout: {{n, strideW * w + dilationW * kw, c}, {kw, c}, {n, w, c}}
3406 /// TODO: w (resp. kw) is unrolled when the strideW ( resp. dilationW) is
3450 ((wSize - 1) * strideW + 1) + ((kwSize - 1) * dilationW + 1) - 1,
3484 // Read lhs slice of size {n, w * strideW + kw * dilationW, c} @ [0, 0,
3517 /*offsets=*/ArrayRef<int64_t>{0, w * strideW + kw * dilationW, 0},
3668 /// {{n, strideW * w + dilationW * kw, c}, {kw, c, f}, {n, w, f}}
3677 if (layout({/*lhsIndex*/ {n, strideW * w + dilationW * kw, c},
3686 /// {{n, c, strideW * w + dilationW * kw}, {f, c, kw}, {n, f, w}}
3694 if (layout({/*lhsIndex*/ {n, c, strideW * w + dilationW * kw},
3703 /// {{n, strideW * w + dilationW * kw, c}, {kw}, {n, w, c}} for pooling
3712 if (layout({/*lhsIndex*/ {n, strideW * w + dilationW * kw, c},
3721 /// {{n, c, strideW * w + dilationW * kw}, {kw}, {n, c, w}} for pooling
3729 if (layout({/*lhsIndex*/ {n, c, strideW * w + dilationW * kw},
3738 /// {{n, strideW * w + dilationW * kw, c}, {kw, c}, {n, w, c}}
3749 if (layout({/*lhsIndex*/ {n, strideW * w + dilationW * kw, c},
3764 int strideW, dilationW;