def extract(cls, node): attrs = { 'merge_repeated': bool(onnx_attr(node, 'merge_repeated', 'i', default=1)), } CTCGreedyDecoderSeqLenOp.update_node_stat(node, attrs) return cls.enabled
def extract(cls, node): attrs = { 'merge_repeated': bool(node.pb.attr['merge_repeated'].b), 'output_sparse_format': True, # Special argument for TF CTCGreedyDecoder replacement transformations } CTCGreedyDecoderSeqLenOp.update_node_stat(node, attrs) return cls.enabled
def extract(cls, node): attrs = { 'merge_repeated': bool(node.pb.attr['merge_repeated'].b), } CTCGreedyDecoderSeqLenOp.update_node_stat(node, attrs) return cls.enabled