Skip to content

Commit b5eacf6

Browse files
committed
fix: Ensure scale and zero-point are scalars
1 parent 4bc2be2 commit b5eacf6

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

src/Dialect/ONNX/ONNXOps/OpHelper.cpp

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -803,6 +803,11 @@ bool hasIntegerPowerExponent(ONNXPowOp *op, int64_t &exponentValue) {
803803
getElementAttributeFromONNXValue(dequantizeOp.getXScale());
804804
ElementsAttr zeroPointAttr =
805805
getElementAttributeFromONNXValue(dequantizeOp.getXZeroPoint());
806+
807+
if (!(isScalarConstantTensor(dequantizeOp.getXScale()) &&
808+
isScalarConstantTensor(dequantizeOp.getXZeroPoint())))
809+
return false;
810+
806811
auto x = getScalarValue<double>(xAttr, xAttr.getElementType());
807812
auto scale = getScalarValue<double>(scaleAttr, scaleAttr.getElementType());
808813
auto zeroPoint =

0 commit comments

Comments
 (0)