Extend MO for operation CTCLoss and partly refactor CTCGreedyDecoder (#588)
authorRoman Kazantsev <roman.kazantsev@intel.com>
Mon, 17 Aug 2020 16:19:59 +0000 (19:19 +0300)
committerGitHub <noreply@github.com>
Mon, 17 Aug 2020 16:19:59 +0000 (19:19 +0300)
commit753150642e83ebe9fc4fec40da4c1989904c4c45
tree8ee489b357c001771982919875e42c384305862b
parent66aa119b11ba69977b37d25872951a9e17a3565c
Extend MO for operation CTCLoss and partly refactor CTCGreedyDecoder (#588)

* Extend MO for operation CTCLoss

* Change sequence length format to a mask format

* Add fixes after first-round review

* Add fixes after the second-round review

* Fixing CTCLossPlusCTCGreedyDecoder transformation
model-optimizer/automation/package_BOM.txt
model-optimizer/extensions/front/caffe/ctcgreedydecoder_ext_test.py
model-optimizer/extensions/front/tf/CTCGreedyDecoderReplacement.py [moved from model-optimizer/extensions/front/tf/CTCGreedyDecoder.py with 78% similarity]
model-optimizer/extensions/front/tf/CTCLossReplacement.py [new file with mode: 0644]
model-optimizer/extensions/front/tf/CTCLossReplacement_test.py [new file with mode: 0644]
model-optimizer/extensions/front/tf/CTCLoss_ext.py [new file with mode: 0644]
model-optimizer/extensions/front/tf/sparse_to_dense_replacer.py
model-optimizer/extensions/ops/ctc_greedy_decoder.py
model-optimizer/extensions/ops/ctc_greedy_decoder_test.py
model-optimizer/extensions/ops/ctc_loss.py [new file with mode: 0644]
model-optimizer/extensions/ops/ctc_loss_test.py [new file with mode: 0644]