def extract(cls, node: Node):
     axis = onnx_attr(node, 'axis', 'i', default=None)
     EltwiseNAdd.update_node_stat(node, {'axis': axis})
     return cls.enabled
Ejemplo n.º 2
0
 def extract(cls, node):
     EltwiseNAdd.update_node_stat(node)
     return cls.enabled
Ejemplo n.º 3
0
 def extract(node):
     EltwiseNAdd.update_node_stat(node)
     return __class__.enabled