[CPU] Disabled sequences decomposition for dynamic case (#10710)

This commit is contained in:
Vladislav Golubev
2022-03-01 22:33:42 +03:00
committed by GitHub
parent 81cd9d86d1
commit e75ee60bec

View File

@@ -285,6 +285,10 @@ static void TransformationUpToCPUSpecificOpSet(std::shared_ptr<ngraph::Function>
auto isSequencePrimitiveSupported = [](const_node_ptr &node) -> bool {
const auto& data = node->input(0);
const auto& data_pshape = data.get_partial_shape();
// WA: dynamic shapes make impossible to check seq_len due to shapeOf subgraphs
// but the sequence is still supported in CPU and doesn't need to be decomposed
if (data_pshape.is_dynamic())
return true;
if (data_pshape.rank().is_static() && data_pshape.rank().get_length() > 1 && !data_pshape[1].is_static())
return false;
auto max_seq_len = data.get_shape().at(1);