IntenseWebs/openvino
3
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Updated inter_tool due to IE api was updated (#7217)

Browse Source
This commit is contained in:
Ilya Sharikov 2021-08-25 11:49:42 +03:00 committed by GitHub
parent ac53316f49
commit ac6c58f03a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/conditional_compilation/test_cc.py
View File

@ -36,6 +36,8 @@ def test_cc_collect(test_id, models, openvino_ref, test_info,
prev_result = glob.glob(f"{out / test_id}.pid*.csv")
for path in prev_result:
os.remove(path)
# Create a directory for infer results, if it haven't done before
infer_out_dir.mkdir(parents=True, exist_ok=True)
# run use case
return_code, output = cmd_exec(
[

4
tests/conditional_compilation/tools/infer_tool.py
View File

@ -27,8 +27,8 @@ def input_preparation(net):
"""
feed_dict = {}
for layer_name, layer_data in net.inputs.items():
feed_dict.update({layer_name: np.ones(shape=layer_data.shape)})
for layer_name, layer_data in net.input_info.items():
feed_dict.update({layer_name: np.ones(shape=layer_data.input_data.shape)})
return feed_dict