relu6.py 869 Bytes
Newer Older
openvino-pushbot's avatar
openvino-pushbot committed
1
"""
2
 Copyright (C) 2018-2020 Intel Corporation
openvino-pushbot's avatar
openvino-pushbot committed
3 4 5 6 7 8 9 10 11 12 13 14 15 16 17

 Licensed under the Apache License, Version 2.0 (the "License");
 you may not use this file except in compliance with the License.
 You may obtain a copy of the License at

      http://www.apache.org/licenses/LICENSE-2.0

 Unless required by applicable law or agreed to in writing, software
 distributed under the License is distributed on an "AS IS" BASIS,
 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 See the License for the specific language governing permissions and
 limitations under the License.
"""

from mo.front.extractor import FrontExtractorOp
18
from extensions.ops.activation_ops import ReLU6
openvino-pushbot's avatar
openvino-pushbot committed
19 20 21 22 23 24


class ReLU6FrontExtractor(FrontExtractorOp):
    op = 'ReLU6'
    enabled = True

25 26
    @classmethod
    def extract(cls, node):
27
        ReLU6.update_node_stat(node)
openvino-pushbot's avatar
openvino-pushbot committed
28
        return True