We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent c34bd9a commit ce063a9Copy full SHA for ce063a9
src/qonnx/util/inference_cost.py
@@ -114,7 +114,7 @@ def inference_cost(
114
qnt_node.domain = "qonnx.custom_op.general"
115
model = model.transform(InferShapes())
116
model = model.transform(GiveUniqueParameterTensors())
117
- model = model.transform(InferDataTypes())
+ model = model.transform(InferDataTypes(allow_scaledint_dtypes=True))
118
model = model.transform(FoldConstants(exclude_op_types=[]))
119
model = model.transform(RemoveUnusedTensors())
120
model = model.transform(RemoveStaticGraphInputs())
0 commit comments