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