We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 59b9536 commit f38d2e9Copy full SHA for f38d2e9
mlir-tensorrt/tensorrt/lib/TensorRT/IR/Verification.cpp
@@ -1023,10 +1023,13 @@ LogicalResult tensorrt::ResizeCubicOp::verify() {
1023
return emitOpError(
1024
"does not support resizing on a tensor that has rank < 2");
1025
1026
- for (int64_t i = outputRank - 3; i >= 0; --i)
+ for (int64_t i = outputRank - 3; i >= 0; --i) {
1027
+ if (inputType.isDynamicDim(i) || outputType.isDynamicDim(i))
1028
+ continue;
1029
if (inputType.getDimSize(i) != outputType.getDimSize(i))
1030
1031
"only supports resizing on the innermost 2 dimensions");
1032
+ }
1033
1034
if (getScales().has_value()) {
1035
if (static_cast<int64_t>(getScales().value().size()) != outputRank)
0 commit comments