From b1ab9b966f41c4d1963eecd628a1cbb80a93e8b4 Mon Sep 17 00:00:00 2001 From: rnugmano Date: Fri, 27 Oct 2023 13:50:53 +0400 Subject: [PATCH] changes ie api --- tests/e2e_oss/pipelines/pipeline_templates/infer_templates.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/e2e_oss/pipelines/pipeline_templates/infer_templates.py b/tests/e2e_oss/pipelines/pipeline_templates/infer_templates.py index 3db040ea5960b8..c38f7f2356815a 100644 --- a/tests/e2e_oss/pipelines/pipeline_templates/infer_templates.py +++ b/tests/e2e_oss/pipelines/pipeline_templates/infer_templates.py @@ -1,7 +1,7 @@ def common_infer_step(device, batch, api_2, input_file_path=None, skip_ir_generation=False, **additional_args): if api_2: # return 'reshape' method for setting batch because old 'batch_size' method still not implemented in new API - infer_api = {"ie_sync": {"device": device, + infer_api = {"ie_sync_api_2": {"device": device, "network_modifiers": {"set_batch_using_reshape_api_2": {"batch": batch}}, **additional_args}} elif not api_2 and skip_ir_generation: