diff options
Diffstat (limited to 'mlir/lib/Dialect/Tosa/Transforms/TosaDecomposeDepthwise.cpp')
| -rw-r--r-- | mlir/lib/Dialect/Tosa/Transforms/TosaDecomposeDepthwise.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/mlir/lib/Dialect/Tosa/Transforms/TosaDecomposeDepthwise.cpp b/mlir/lib/Dialect/Tosa/Transforms/TosaDecomposeDepthwise.cpp index e6fba211dc37..14f392ab8c45 100644 --- a/mlir/lib/Dialect/Tosa/Transforms/TosaDecomposeDepthwise.cpp +++ b/mlir/lib/Dialect/Tosa/Transforms/TosaDecomposeDepthwise.cpp @@ -108,7 +108,7 @@ struct DepthwiseConv2DIsMul : public OpRewritePattern<tosa::DepthwiseConv2DOp> { } } - auto padSizeTy = RankedTensorType::get({5, 2}, rewriter.getI64Type()); + auto padSizeTy = RankedTensorType::get({10}, rewriter.getI64Type()); auto padSize = DenseIntElementsAttr::get(padSizeTy, ArrayRef<int64_t>(pad)); Value padSizeVal = |