def extract(cls, node):
Sub.update_node_stat(node, {'data_type': tf_dtype_extractor(node.pb.attr["T"].type)})
return cls.enabled
def extract(cls, node: Node):
axis = onnx_attr(node, 'axis', 'i', default=None)
Sub.update_node_stat(node, {'axis': axis})
return cls.enabled
def extract(cls, node):
Sub.update_node_stat(node)
return cls.enabled