We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 97299fe commit 2234272Copy full SHA for 2234272
docs/articles_en/assets/snippets/ov_extensions.py
@@ -47,7 +47,7 @@ def conversion(node):
47
#! [py_frontend_extension_aten_hardtanh]
48
import torch
49
from openvino.frontend import ConversionExtension, NodeContext
50
-from openvino.tools.mo import convert_model
+from openvino import convert_model
51
52
53
class HardTanh(torch.nn.Module):
@@ -69,5 +69,5 @@ def convert_hardtanh(node: NodeContext):
69
70
model = HardTanh(min_val=0.1, max_val=2.0)
71
hardtanh_ext = ConversionExtension("aten::hardtanh", convert_hardtanh)
72
-ov_model = convert_model(input_model=model, extensions=[hardtanh_ext])
+ov_model = convert_model(input_model=model, extension=[hardtanh_ext])
73
0 commit comments