Written reading the operation CTCGreedyDecoder for ONNX (as a custom operation). (#6215)

This commit is contained in:
Vladimir Gavrilov
2021-06-18 09:54:41 +03:00
committed by GitHub
parent 8543b42ac0
commit 70f9d8564e
8 changed files with 29 additions and 9 deletions
@@ -3,7 +3,7 @@
import unittest
from extensions.front.tf.CTCGreedyDecoderReplacement import CTCGreedyDecoderReplacement, CTCGreedyDecoderWithSparseToDenseShapeReplacement
from extensions.front.CTCGreedyDecoderReplacement import CTCGreedyDecoderReplacement, CTCGreedyDecoderWithSparseToDenseShapeReplacement
from mo.front.common.partial_infer.utils import int64_array
from mo.utils.ir_engine.compare_graphs import compare_graphs
from unit_tests.utils.graph import build_graph, const
@@ -5,7 +5,7 @@ import numpy as np
import unittest
from argparse import Namespace
from extensions.front.tf.CTCLossReplacement import CTCLossReplacement
from extensions.front.CTCLossReplacement import CTCLossReplacement
from mo.front.common.partial_infer.utils import int64_array
from mo.utils.ir_engine.compare_graphs import compare_graphs
from unit_tests.utils.graph import build_graph, const
@@ -3,7 +3,7 @@
import unittest
from extensions.front.tf.sparse_to_dense_replacer import SparseToDenseReplacer
from extensions.front.sparse_to_dense_replacer import SparseToDenseReplacer
from mo.front.common.partial_infer.utils import int64_array
from mo.utils.ir_engine.compare_graphs import compare_graphs
from unit_tests.utils.graph import build_graph, const