Searched defs:dilationAttr (Results 1 – 1 of 1) sorted by relevance
/llvm-project/mlir/lib/Conversion/TosaToLinalg/ |
H A D | TosaToLinalgNamed.cpp | 156 getConvOrPoolOutputDim(Location loc,Value inputDim,int64_t padBeforeAttr,int64_t padAfterAttr,Value kernelDim,int64_t strideAttr,int64_t dilationAttr,OpBuilder & rewriter) getConvOrPoolOutputDim() argument 181 inferDynamicDimsForConv(Location loc,Value input,Value weight,ShapedType resultTy,ArrayRef<int64_t> padAttr,ArrayRef<int64_t> strideAttr,ArrayRef<int64_t> dilationAttr,ArrayRef<int64_t> inputSizeDims,ArrayRef<int64_t> kernelSizeDims,OpBuilder & rewriter) inferDynamicDimsForConv() argument 363 auto dilationAttr = rewriter.getI64TensorAttr(dilation); matchAndRewrite() local 485 auto dilationAttr = rewriter.getI64TensorAttr(dilation); matchAndRewrite() local 783 Attribute dilationAttr = rewriter.getI64VectorAttr({1, 1}); matchAndRewrite() local 844 Attribute dilationAttr = rewriter.getI64VectorAttr({1, 1}); matchAndRewrite() local [all...] |