예제 #1
0
 def extract(cls, node):
     exclusive = node.pb.attr['exclusive'].b
     reverse = node.pb.attr['reverse'].b
     CumSum.update_node_stat(node, {
         'exclusive': exclusive,
         'reverse': reverse
     })
     return cls.enabled
 def extract(cls, node):
     exclusive = onnx_attr(node, 'exclusive', 'i', 0)
     reverse = onnx_attr(node, 'reverse', 'i', 0)
     CumSum.update_node_stat(node, {
         'exclusive': exclusive,
         'reverse': reverse
     })
     return cls.enabled