-
Notifications
You must be signed in to change notification settings - Fork 74k
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
Modify Transpose kernel to work in TFLu #47446
Modify Transpose kernel to work in TFLu #47446
Conversation
Thanks for contributing to TensorFlow Lite Micro. To keep this process moving along, we'd like to make sure that you have completed the items on this list:
We would like to have a discussion on the Github issue first to determine the best path forward, and then proceed to the PR review. |
This PR also impacts the performance changes on the TFLite. Could you use the micro macro to divide the logics? For example, keeping the original logic for the TFlite and using a new logics for only micro cases. |
@abattery I am not sure I understand what you mean. The changes in this PR should only affect the micro kernel. How does this impact the performance changes on TFLite? |
Sorry @patriklaurell you are right. I misread the filename. Please ignore my comment. |
Hi, which is the status of this PR? Thank you. |
60ce10d
to
774b573
Compare
@petewarden I updated this PR and added two commits for porting the tests and adding transpose to the micro build. It is ready for review. |
@advaitjain @petewarden The MacOS CPU Python3 test is failing but there is very little information in the test log and I have no way of reproducing the error locally. Do you have any insight into why it fails? |
Unfortunately, I don't know why it is failing either. Let's have you address the review comments and then we can try and figure out what is going on with the MacOS build. |
@advaitjain ready for review. Did you find anything regarding the MacOS test? |
I really do not know what is going on with the MacOS failure. @petewarden, any ideas? |
Can you merge master and then get this PR building again after changes from 822ddbf That will fix the MacOS and internal CI errors. |
This PR modifies the transpose kernel to make it run in micro, ports the tests and adds the kernel to the micro build.
This is PR 4/4 in delivering #45695