Skip to content

Commit

Permalink
Update test_float16.py
Browse files Browse the repository at this point in the history
  • Loading branch information
xiaowuhu committed May 24, 2024
1 parent 37ab3f2 commit 8396a63
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion tests/test_float16.py
Original file line number Diff line number Diff line change
Expand Up @@ -43,7 +43,6 @@ def transpose_n_matmul(x):
m1 = np.array([[2, 3], [4, 5], [6, 7]]).astype(np.float32).reshape([1, 1, 6, 1])
expected = transpose_n_matmul(m1)
model = transpose_n_matmul.to_model()
onnx.save_model(model, 'd:/float32.onnx')
actual = _ort_inference(model, {'x': m1})
self.assertTrue(np.allclose(expected, actual))

Expand Down

0 comments on commit 8396a63

Please sign in to comment.