Skip to content
Snippets Groups Projects
Commit b3fabb52 authored by Yaman Umuroglu's avatar Yaman Umuroglu
Browse files

[Test] fix test cond in infer_datatypes

parent e1b4bb52
No related branches found
No related tags found
No related merge requests found
......@@ -54,8 +54,8 @@ def test_infer_datatypes():
assert model.get_tensor_datatype("MatMul_1_out0") == DataType.INT32
assert model.get_tensor_datatype("MatMul_2_out0") == DataType.INT32
assert model.get_tensor_datatype("MatMul_3_out0") == DataType.INT32
assert model.get_tensor_datatype("Sign_0_out0") == DataType.BIPOLAR
assert model.get_tensor_datatype("Sign_1_out0") == DataType.BIPOLAR
assert model.get_tensor_datatype("Sign_2_out0") == DataType.BIPOLAR
assert model.get_tensor_datatype("Sign_3_out0") == DataType.BIPOLAR
assert model.get_tensor_datatype("MultiThreshold_0_out0") == DataType.BIPOLAR
assert model.get_tensor_datatype("MultiThreshold_1_out0") == DataType.BIPOLAR
assert model.get_tensor_datatype("MultiThreshold_2_out0") == DataType.BIPOLAR
assert model.get_tensor_datatype("MultiThreshold_3_out0") == DataType.BIPOLAR
os.remove(export_onnx_path)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment