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