return IENetwork instead of Function

openvinotoolkit · Aug 29, 2023 · 7a7f1db · 7a7f1db
1 parent c235196
commit 7a7f1db
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 10 deletions.
diff --git a/docs/snippets/ov_preprocessing_migration.py b/docs/snippets/ov_preprocessing_migration.py
@@ -71,9 +71,7 @@
 operation_name = "data"
 
 core = ie.IECore()
-func = get_ngraph_model()
-caps = ng.Function.to_capsule(func)
-network = ie.IENetwork(caps)
+network = get_ngraph_model()
 
 
 #! [mean_scale]

diff --git a/docs/snippets/ov_properties_migration.py b/docs/snippets/ov_properties_migration.py
@@ -6,7 +6,7 @@
 import openvino.runtime.properties as props
 import openvino.runtime.properties.hint as hints
 
-from utils import get_model, get_ngraph_model
+from utils import get_model
 
 core = ov.Core()
 
@@ -63,9 +63,7 @@
 core.set_config({"PERF_COUNT": "YES"}, "CPU")
 #! [core_set_config]
 
-func = get_ngraph_model()
-caps = ng.Function.to_capsule(func)
-net = ie.IENetwork(caps)
+net = get_ngraph_model()
 
 #! [core_load_network]
 exec_network = core.load_network(net, "MULTI", {"DEVICE_PRIORITIES": "CPU, GPU",

diff --git a/docs/snippets/utils.py b/docs/snippets/utils.py
@@ -36,10 +36,12 @@ def get_ngraph_model(input_shape = None, input_dtype=np.float32):
         input_shape = [1, 3, 32, 32]
     param = ng.opset11.parameter(input_shape, input_dtype, name="data")
     relu = ng.opset11.relu(param, name="relu")
-    model = Function([relu], [param], "test_model")
+    func = Function([relu], [param], "test_model")
+    caps = ng.Function.to_capsule(func)
+    net = ie.IENetwork(caps)
 
-    assert model is not None
-    return model
+    assert net is not None
+    return net
 
 
 def get_image(shape = (1, 3, 32, 32), dtype = "float32"):