Files
openvino/model-optimizer/mo/back/replacement.py
T
Alexey Suhov 6478f1742a Align copyright notice in python scripts (CVS-51320) (#4974)
* Align copyright notice in python scripts (CVS-51320)
2021-03-26 17:54:28 +03:00

26 lines
686 B
Python

# Copyright (C) 2018-2021 Intel Corporation
# SPDX-License-Identifier: Apache-2.0
from mo.utils import class_registration
from mo.utils.replacement_pattern import ReplacementPattern
class BackReplacementPattern(ReplacementPattern):
registered_ops = {}
registered_cls = []
def run_after(self):
from extensions.back.pass_separator import BackStart
return [BackStart]
def run_before(self):
from extensions.back.pass_separator import BackFinish
return [BackFinish]
@classmethod
def class_type(cls):
return class_registration.ClassType.BACK_REPLACER
ReplacementPattern.excluded_replacers.append(BackReplacementPattern)