Skip to content

Commit

Permalink
only lstmseq decomposition
Browse files Browse the repository at this point in the history
  • Loading branch information
michal-miotk committed Oct 28, 2024
1 parent 63bddcb commit 2b6ad65
Showing 1 changed file with 1 addition and 4 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -517,10 +517,7 @@ void TransformationsPipeline::apply(std::shared_ptr<ov::Model> func) {
} else if (std::dynamic_pointer_cast<const ov::op::v5::GRUSequence>(node)) {
return false;
} else if (const auto &lstm_seq = std::dynamic_pointer_cast<const ov::op::v5::LSTMSequence>(node)) {
return lstm_seq->get_clip() == 0.0f &&
lstm_seq->get_activations() == std::vector<std::string>{"sigmoid", "tanh", "tanh"} &&
!ov::op::util::is_seq_len_provided(lstm_seq->get_input_node_shared_ptr(0),
lstm_seq->get_input_node_shared_ptr(3));
return false;
}
return false;
};
Expand Down

0 comments on commit 2b6ad65

Please sign in to comment.