def extract(node: Node): attr = get_mxnet_layer_attrs(node.symbol_dict) node_attrs = { 'epsilon': attr.float('eps', 0.001) } InstanceNormalization.update_node_stat(node, node_attrs) return __class__.enabled
def extract(node): epsilon = onnx_attr(node, 'epsilon', 'f', default=float(1e-5)) InstanceNormalization.update_node_stat(node, {'epsilon': epsilon}) return __class__.enabled