Commit f1431da 1 parent 9201c57 commit f1431da Copy full SHA for f1431da
File tree 1 file changed +2
-2
lines changed
docs/articles_en/assets/snippets
1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -47,7 +47,7 @@ def conversion(node):
47
47
#! [py_frontend_extension_aten_hardtanh]
48
48
import torch
49
49
from openvino .frontend import ConversionExtension , NodeContext
50
- from openvino . tools . mo import convert_model
50
+ from openvino import convert_model
51
51
52
52
53
53
class HardTanh (torch .nn .Module ):
@@ -69,5 +69,5 @@ def convert_hardtanh(node: NodeContext):
69
69
70
70
model = HardTanh (min_val = 0.1 , max_val = 2.0 )
71
71
hardtanh_ext = ConversionExtension ("aten::hardtanh" , convert_hardtanh )
72
- ov_model = convert_model (input_model = model , extensions = [hardtanh_ext ])
72
+ ov_model = convert_model (input_model = model , extension = [hardtanh_ext ])
73
73
#! [py_frontend_extension_aten_hardtanh]
You can’t perform that action at this time.
0 commit comments