Skip to content

Comments

Refactor conv1d call to ensure consistent state handling#34

Open
gukgurgel wants to merge 1 commit intoNX-AI:mainfrom
gukgurgel:b
Open

Refactor conv1d call to ensure consistent state handling#34
gukgurgel wants to merge 1 commit intoNX-AI:mainfrom
gukgurgel:b

Conversation

@gukgurgel
Copy link

  • Switched the assignment of x_conv and conv_state based on return_last_state condition
  • Ensured conv_state is assigned when return_last_state is True

- Switched the assignment of `x_conv` and `conv_state` based on `return_last_state` condition
- Ensured `conv_state` is assigned when `return_last_state` is True

Signed-off-by: Gustavo Korzune Gurgel <gus.gurgel@tum.de>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant