Exemplo n.º 1
0
 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
Exemplo n.º 2
0
 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