Skip to content

Commit 94c2939

Browse files
authored
Sync from tflite-micro. (tensorflow#178)
1 parent 285b5a0 commit 94c2939

File tree

3 files changed

+8
-11
lines changed

3 files changed

+8
-11
lines changed

src/tensorflow/lite/micro/kernels/micro_ops.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -94,6 +94,7 @@ TfLiteRegistration Register_REDUCE_MAX();
9494
TfLiteRegistration Register_RELU();
9595
TfLiteRegistration Register_RELU6();
9696
TfLiteRegistration Register_RESIZE_BILINEAR();
97+
TfLiteRegistration Register_RESIZE_NEAREST_NEIGHBOR();
9798
TfLiteRegistration Register_SELECT_V2();
9899
TfLiteRegistration Register_SHAPE();
99100
TfLiteRegistration Register_SLICE();
@@ -124,7 +125,6 @@ TfLiteRegistration Register_COS();
124125
TfLiteRegistration Register_LOG();
125126
TfLiteRegistration Register_LOGICAL_NOT();
126127
TfLiteRegistration Register_RESHAPE();
127-
TfLiteRegistration Register_RESIZE_NEAREST_NEIGHBOR();
128128
TfLiteRegistration Register_ROUND();
129129
TfLiteRegistration Register_RSQRT();
130130
TfLiteRegistration Register_SIN();

src/tensorflow/lite/micro/kernels/resize_nearest_neighbor.cpp

Lines changed: 6 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
/* Copyright 2017 The TensorFlow Authors. All Rights Reserved.
1+
/* Copyright 2023 The TensorFlow Authors. All Rights Reserved.
22
33
Licensed under the Apache License, Version 2.0 (the "License");
44
you may not use this file except in compliance with the License.
@@ -24,9 +24,8 @@ limitations under the License.
2424
#include "tensorflow/lite/micro/micro_log.h"
2525

2626
namespace tflite {
27-
namespace ops {
28-
namespace micro {
29-
namespace resize_nearest_neighbor {
27+
28+
namespace {
3029

3130
constexpr int kInputTensor = 0;
3231
constexpr int kSizeTensor = 1;
@@ -114,13 +113,11 @@ TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) {
114113

115114
return kTfLiteOk;
116115
}
117-
} // namespace resize_nearest_neighbor
116+
117+
} // namespace
118118

119119
TfLiteRegistration Register_RESIZE_NEAREST_NEIGHBOR() {
120-
return tflite::micro::RegisterOp(nullptr, resize_nearest_neighbor::Prepare,
121-
resize_nearest_neighbor::Eval);
120+
return tflite::micro::RegisterOp(nullptr, Prepare, Eval);
122121
}
123122

124-
} // namespace micro
125-
} // namespace ops
126123
} // namespace tflite

src/tensorflow/lite/micro/micro_mutable_op_resolver.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -444,7 +444,7 @@ class MicroMutableOpResolver : public MicroOpResolver {
444444

445445
TfLiteStatus AddResizeNearestNeighbor() {
446446
return AddBuiltin(BuiltinOperator_RESIZE_NEAREST_NEIGHBOR,
447-
tflite::ops::micro::Register_RESIZE_NEAREST_NEIGHBOR(),
447+
Register_RESIZE_NEAREST_NEIGHBOR(),
448448
ParseResizeNearestNeighbor);
449449
}
450450

0 commit comments

Comments
 (0)