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 5f5dc24 commit 939afb0Copy full SHA for 939afb0
modelopt/onnx/autocast/precisionconverter.py
@@ -53,7 +53,7 @@
53
OP_TYPES_NOT_SUPPORTED_IN_LOW_PRECISION.extend(["Scan", "If", "Loop", "LSTM"])
54
55
# Mapping of op types to indices of inputs that should not be converted to low precision.
56
-SKIP_LOW_PRECISION_MAPPING_FP16 = {"Resize": {1}}
+SKIP_LOW_PRECISION_MAPPING_FP16 = {"Resize": {2}}
57
SKIP_LOW_PRECISION_MAPPING_BF16 = {"Resize": {1, 2}}
58
59
0 commit comments