Skip to content

Commit 1da8669

Browse files
committed
fix : test code fix
1 parent 7cfa563 commit 1da8669

File tree

1 file changed

+3
-12
lines changed

1 file changed

+3
-12
lines changed

tensorflow/lite/micro/one_hot_test.cc

Lines changed: 3 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -4,15 +4,6 @@
44
#include "tensorflow/lite/micro/test_helpers.h"
55
#include "tensorflow/lite/micro/testing/micro_test.h"
66

7-
namespace tflite {
8-
namespace ops {
9-
namespace micro {
10-
11-
const TFLMRegistration* Register_ONE_HOT();
12-
} // namespace micro
13-
} // namespace ops
14-
} // namespace tflite
15-
167
namespace tflite {
178
namespace testing {
189
namespace {
@@ -52,9 +43,9 @@ void TestOneHot(const int* indices_dims, const int32_t* indices_data,
5243
int outputs_array_data[] = {1, 4}; // output tensor index
5344
TfLiteIntArray* outputs_array = IntArrayFromInts(outputs_array_data);
5445

55-
// tflite::ops::micro::Register_ONE_HOT)
56-
const TFLMRegistration registration = *tflite::ops::micro::Register_ONE_HOT();
57-
micro::KernelRunner runner(registration, tensors, tensors_size, inputs_array,
46+
// tflite::Register_ONE_HOT)
47+
const TFLMRegistration* registration = tflite::Register_ONE_HOT();
48+
micro::KernelRunner runner(*registration, tensors, tensors_size, inputs_array,
5849
outputs_array,
5950
reinterpret_cast<void*>(&builtin_data));
6051

0 commit comments

Comments
 (0)