def extract(cls, node):
attrs = get_mxnet_layer_attrs(node.symbol_dict)
ReduceMax.update_node_stat(node, {
'axis': int64_array([attrs.int('axis', 0)]),
'keep_dims': False
})
return cls.enabled
def extract(cls, node: Node):
axis = onnx_attr(node,
'axes',
'ints',
default=None,
dst_type=lambda x: int64_array(x))
keep_dims = onnx_attr(node, 'keepdims', 'i', default=True)
ReduceMax.update_node_stat(node, {
'axis': axis,
'keep_dims': keep_dims
})
return cls.enabled
def extract(cls, node: Node):
ReduceMax.update_node_stat(node,
{'keep_dims': node.pb.attr['keep_dims'].b})
return cls.enabled