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

Any possibility of supporting transpose operation? #193

Closed
ppershing opened this issue Aug 10, 2020 · 5 comments
Closed

Any possibility of supporting transpose operation? #193

ppershing opened this issue Aug 10, 2020 · 5 comments

Comments

@ppershing
Copy link

According to https://coral.ai/docs/edgetpu/models-intro/#supported-operations coral does not seem to support Transpose op. I would presume that under some limiting circumstances (e.g., not transposing batch dim, etc.) the device should be able to support this. And the operation would be very helpful -- given size limitations, using grouped convolutions might be an easy way to get better accuracy. However, such architectures usually involve a channel "shuffle" so that channels from different groups can get mixed and such shuffle is usually implemented by reshape + transpose + reshape.

@nesherofek
Copy link

+1

@Namburger
Copy link

Hi guys, I do have a feature request opened for this internally, will keep update if we move forward with it.

@nesherofek
Copy link

nesherofek commented Oct 29, 2020

For any one face this issue
by converting pytorch to keras (using this rep https://github.com/nerox8664/pytorch2keras ) i was able to fully convert my model to NHWC and by that avoiding the TRANSPOE nodes, edge tpu comile was able to map all operations to TPU (:

@MekhailS
Copy link

MekhailS commented Aug 3, 2021

+1, badly need this feature

@manoj7410
Copy link

Compiler version 16 supports 'Transpose' operations. Please see the list of supported operations at https://coral.ai/docs/edgetpu/models-intro/#supported-operations.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

No branches or pull requests

5 participants