Files
openvino/model-optimizer/mo/front/mxnet/extractors/l2_normalization.py
T

19 lines
439 B
Python
Raw Normal View History

# Copyright (C) 2018-2021 Intel Corporation
# SPDX-License-Identifier: Apache-2.0
2018-10-16 13:45:03 +03:00
from mo.front.common.partial_infer.elemental import copy_shape_infer
def l2_normalization_ext(attrs):
eps = attrs.float('eps', 1e-10)
node_attrs = {
2018-11-23 16:19:43 +03:00
'op': 'Normalize',
2018-10-16 13:45:03 +03:00
'type': 'Normalize',
'eps': eps,
'across_spatial': 0,
'channel_shared': 0,
'infer': copy_shape_infer
}
return node_attrs