def extract(cls, node):
pb = node.pb
mapping_rule = {
'axis': pb.concat_param.axis,
}
Concat.update_node_stat(node, mapping_rule)
return cls.enabled
def extract(cls, node):
attrs = get_mxnet_layer_attrs(node.symbol_dict)
data = {
'axis': attrs.int("dim", 1),
}
# update the attributes of the node
Concat.update_node_stat(node, data)
return cls.enabled
def extract(node):
attrs = {'N': node.pb.attr["N"].i, 'simple_concat': True}
Concat.update_node_stat(node, attrs)
return __class__.enabled
def extract(cls, node):
attrs = {
'axis': node.module.dim,
}
Concat.update_node_stat(node, attrs)
return cls.enabled
def extract(cls, node):
mapping_rule = {'axis': 1}
Concat.update_node_stat(node, mapping_rule)
return cls.enabled
def extract(cls, node):
mapping_rule = {'axis': onnx_attr(node, 'axis', 'i', default=0)}
Concat.update_node_stat(node, mapping_rule)
return cls.enabled
def extract(node):
mapping_rule = {
'axis': 1
}
Concat.update_node_stat(node, mapping_rule)
return __class__.enabled