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

[Test] also retrieve full dict for golden end2end v test

parent 3dcc3dd5
No related branches found
No related tags found
No related merge requests found
......@@ -168,7 +168,8 @@ def test_end2end_tfc_verify_all():
oname = golden.graph.output[0].name
ishape = golden.get_tensor_shape(iname)
x = np.zeros(ishape, dtype=np.float32)
y_golden = execute_onnx(golden, {iname: x})[oname]
ret_golden = execute_onnx(golden, {iname: x}, True)
y_golden = ret_golden[oname]
# set up parent+child graph to test
# we'll use models from the previous step as the child model
parent_model = ModelWrapper(build_dir + "/end2end_tfc_w1_a1_dataflow_parent.onnx")
......
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