示例#1
0
 def extract(cls, node):
     Sub.update_node_stat(
         node, {'data_type': tf_dtype_extractor(node.pb.attr["T"].type)})
     return cls.enabled
示例#2
0
 def extract(cls, node: Node):
     axis = onnx_attr(node, 'axis', 'i', default=None)
     Sub.update_node_stat(node, {'axis': axis})
     return cls.enabled
示例#3
0
 def extract(cls, node):
     Sub.update_node_stat(node)
     return cls.enabled
示例#4
0
 def extract(node):
     Sub.update_node_stat(node)
     return __class__.enabled