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