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

[Feature Request] Add support for ttnn.flip #16618

Open
sankarmanoj-tt opened this issue Jan 10, 2025 · 1 comment
Open

[Feature Request] Add support for ttnn.flip #16618

sankarmanoj-tt opened this issue Jan 10, 2025 · 1 comment
Assignees
Labels
feature-request External feature request op_cat: TM

Comments

@sankarmanoj-tt
Copy link
Contributor

Describe the solution you'd like
Have a ttnn equivalent of torch.flip

Describe alternatives you've considered
Currently, the flip operation is implemented on the host, which is very slow.

Additional context
This is necessary for loading the weights for conv_transpose2d.

@pavlejosipovic
Copy link
Contributor

Seems like flip would be useful for backward Conv as well.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
feature-request External feature request op_cat: TM
Projects
None yet
Development

No branches or pull requests

4 participants