-
Notifications
You must be signed in to change notification settings - Fork 968
TFLM: Add ONE_HOT operator and unit tests #3260
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
base: main
Are you sure you want to change the base?
TFLM: Add ONE_HOT operator and unit tests #3260
Conversation
|
Thanks for your pull request! It looks like this may be your first contribution to a Google open source project. Before we can look at your pull request, you'll need to sign a Contributor License Agreement (CLA). View this failed invocation of the CLA check for more information. For the most up to date status, view the checks section at the bottom of the pull request. |
|
@googlebot I signed it! |
26628b8 to
8ecac95
Compare
|
@junseokShim Please add a line at the end of your PR description above: |
|
@junseokShim This does not seem to be a completed PR. Please adhere to the requirements as listed in issue #3078. |
OK, I will double-check the detailed requirements and make sure they are reflected. |
This PR adds ONE_HOT operator support to TensorFlow Lite Micro.
tensorflow/lite/micro/kernels/one_hot.cctensorflow/lite/micro/kernels/one_hot.htensorflow/lite/micro/one_hot_test.ccmake -f tensorflow/lite/micro/tools/make/Makefile test_one_hot_testclang-formatandcpplint.pyon modified filesThis intends to address #3078.
bug=fixes #3078