コード例 #1
0
ファイル: elementwise_ext.py プロジェクト: zkzt/openvino 
 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