Skip to content

Commit 26628b8

Browse files
committed
TFLM: Add ONE_HOT operator implementation and tests
1 parent b90467d commit 26628b8

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

tensorflow/lite/micro/one_hot_test.cc

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -71,6 +71,7 @@ void TestOneHot(const int* indices_dims, const int32_t* indices_data,
7171
} // namespace testing
7272
} // namespace tflite
7373

74+
// UNIT TEST
7475
TF_LITE_MICRO_TESTS_BEGIN
7576

7677
TF_LITE_MICRO_TEST(OneHot_BasicInt32) {

0 commit comments

Comments
 (0)