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

MO: --transform Pruning option (#9304)

Browse Source
This commit is contained in:
Gleb Kazantaev
2021-12-20 15:55:50 +03:00
committed by GitHub
parent 191b15a228
commit 249e1266fb
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
tools/mo/openvino/tools/mo/back/offline_transformations.py
View File

@@ -9,10 +9,13 @@ from openvino.tools.mo.utils.cli_parser import parse_transform
def get_available_transformations():
try:
from openvino.offline_transformations_pybind import apply_low_latency_transformation, apply_make_stateful_transformation # pylint: disable=import-error,no-name-in-module
from openvino.offline_transformations_pybind import apply_low_latency_transformation # pylint: disable=import-error,no-name-in-module
from openvino.offline_transformations_pybind import apply_make_stateful_transformation # pylint: disable=import-error,no-name-in-module
from openvino.offline_transformations_pybind import apply_pruning_transformation # pylint: disable=import-error,no-name-in-module
return {
'MakeStateful': apply_make_stateful_transformation,
'LowLatency2': apply_low_latency_transformation,
'Pruning': apply_pruning_transformation,
}
except Exception as e:
return {}