Lines Matching refs:stride
99 llvm::SmallVector<int64_t> stride; in matchAndRewrite() local
102 getValuesFromIntArrayAttribute(op.getStride().cast<ArrayAttr>(), stride); in matchAndRewrite()
107 if (llvm::any_of(stride, [](int64_t v) { return v != 1; })) in matchAndRewrite()
132 rewriter.getI64ArrayAttr(convPad), rewriter.getI64ArrayAttr(stride), in matchAndRewrite()
137 rewriter.getI64ArrayAttr(convPad), rewriter.getI64ArrayAttr(stride), in matchAndRewrite()
168 llvm::SmallVector<int64_t> stride; in matchAndRewrite() local
171 getValuesFromIntArrayAttribute(op.getStride().cast<ArrayAttr>(), stride); in matchAndRewrite()
178 if (llvm::all_of(stride, [](int64_t v) { return v == 1; })) in matchAndRewrite()
195 weightHeight % stride[0] ? stride[0] - weightHeight % stride[0] : 0; in matchAndRewrite()
197 weightWidth % stride[1] ? stride[1] - weightWidth % stride[1] : 0; in matchAndRewrite()
222 outputChannels, weightHeight / stride[0], in matchAndRewrite()
223 stride[0], weightWidth / stride[1], in matchAndRewrite()
224 stride[1], inputChannels}; in matchAndRewrite()
240 outputChannels * stride[0] * stride[1], weightHeight / stride[0], in matchAndRewrite()
241 weightWidth / stride[1], inputChannels}; in matchAndRewrite()
282 RankedTensorType::get({outputChannels * stride[0] * stride[1]}, in matchAndRewrite()
285 RankedTensorType::get({outputChannels * stride[0] * stride[1]}, in matchAndRewrite()
319 batch, convHeight, convWidth, stride[0], stride[1], outputChannels}; in matchAndRewrite()
335 batch, convHeight * stride[0], convWidth * stride[1], outputChannels}; in matchAndRewrite()