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 55c2f6f commit a3451c5Copy full SHA for a3451c5
tests/transformation/test_conv_lowering.py
@@ -55,7 +55,7 @@ def test_conv_lowering_quant_weights(model_name):
55
input_dict = {model.graph.input[0].name: input_t}
56
prod_dict = oxe.execute_onnx(model, input_dict)
57
prod_t = prod_dict[model.graph.output[0].name]
58
- assert (prod_t == golden_t).all()
+ assert np.isclose(prod_t, golden_t).all()
59
60
61
def test_conv_lowering_convmnist():
0 commit comments