openvino/model-optimizer/extensions/front/mxnet/null_ext.py

20 lines
545 B
Python

# Copyright (C) 2018-2021 Intel Corporation
# SPDX-License-Identifier: Apache-2.0
from extensions.ops.parameter import Parameter
from mo.front.extractor import FrontExtractorOp
from mo.ops.const import Const
class NullFrontExtractor(FrontExtractorOp):
op = 'null'
enabled = True
@classmethod
def extract(cls, node):
if 'value' in node.symbol_dict:
Const.update_node_stat(node, {'value': node.symbol_dict['value']})
else:
Parameter.update_node_stat(node, {})
return cls.enabled