Skip to content

Commit 7e22c82

Browse files
slyubomirskyyuanfz98
authored andcommitted
[frontend][keras] Add support for TimeDistributed (apache#7006)
* First pass on modifying Keras importer to handle TimeDistributed * Use squeeze inside TimeDistributed, add tests * linter fixes * More linting * Even more linting * Fix unused argument annotations * Forgot one pylint annotation * Forgot to set up data layout in _convert_activation * Decouple data_layout from etab * Linting fix * Forgot to set data_layout argument * Missed an etab.data_format, also test_conv1d was not in the test file's main * Rebase fixes * Linting fix * _convert_lambda needs a data layout argument too * linting fix too * Lint the test file too * Redundant variables * Simplify further * Another simplification Co-authored-by: Steven Lyubomirsky <slyubomirsky@octoml.ai>
1 parent d759a72 commit 7e22c82

2 files changed

Lines changed: 264 additions & 100 deletions

File tree

0 commit comments

Comments
 (0)