Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[MO] Support TF2 Keras ConvLSTM2D operation #4197

Conversation

rkazants
Copy link
Contributor

@rkazants rkazants commented Feb 5, 2021

Description: it implements support of TF2 Keras ConvLSTM2D by the MO tool and adds transformation for pulling constant nodes into a body graph of Loop node. Inference results match.

Ticket: 47278

Code:

  • Comments
  • Code style (PEP8)
  • Transformation generates reshape-able IR
  • Transformation preserves original framework node names

Validation:

  • Unit tests - we have a separate testing activity
  • Framework operation tests - we have a separate testing activity
  • Transformation tests - we have a separate testing activity
  • e2e model test with an update of ./tests/e2e_oss/utils/reshape_utils.py - N/A, no model
  • Model Optimizer IR Reader check

Documentation:

  • Supported frameworks operations list - we have a separate documentation activity
  • Supported public models list - N/A
  • New operations specification - N/A
  • Guide on how to convert the public model - N/A
  • User guide update - N/A

Signed-off-by: Roman Kazantsev [email protected]

@rkazants rkazants requested a review from a team February 5, 2021 09:52
@rkazants rkazants force-pushed the feature/rkazants/tf2_keras_convlstm2d branch from df131b2 to e03e90c Compare February 5, 2021 09:56
@openvino-pushbot openvino-pushbot added the category: MO Model Optimizer label Feb 5, 2021
@rkazants rkazants force-pushed the feature/rkazants/tf2_keras_convlstm2d branch from e03e90c to 2d1111b Compare February 8, 2021 13:50
@rkazants rkazants force-pushed the feature/rkazants/tf2_keras_convlstm2d branch from 2d1111b to b261746 Compare February 8, 2021 14:08
@rkazants rkazants requested review from a team, mvafin, iimironov, vgavrilo and lazarevevgeny and removed request for a team February 8, 2021 15:17
@rkazants rkazants changed the title Support TF2 Keras ConvLSTM2D operation [MO] Support TF2 Keras ConvLSTM2D operation Feb 8, 2021
@rkazants rkazants requested a review from iimironov February 8, 2021 18:55
@lazarevevgeny lazarevevgeny merged commit 636f5c4 into openvinotoolkit:master Feb 9, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
category: MO Model Optimizer
Projects
None yet
Development

Successfully merging this pull request may close these issues.

6 participants