Refactoring: Make OpResolver return const pointer.

PiperOrigin-RevId: 196587227

1 files changed, 4 insertions, 6 deletions

diff --git a/tensorflow/contrib/lite/kernels/test_util.h b/tensorflow/contrib/lite/kernels/test_util.h
index 32529b6d94..55edc97d19 100644
--- a/tensorflow/contrib/lite/kernels/test_util.h
+++ b/tensorflow/contrib/lite/kernels/test_util.h
@@ -93,16 +93,14 @@ class SingleOpResolver : public OpResolver {
     registration_.builtin_code = static_cast<int32_t>(op);
     registration_.version = 1;
   }
-  TfLiteRegistration* FindOp(BuiltinOperator op, int version) const override {
+  const TfLiteRegistration* FindOp(BuiltinOperator op,
+                                   int version) const override {
     if (op == op_) {
-      // The current interface requires to return a mutable pointer, but the
-      // caller never changes the structure.
-      // TODO(ycling): Consider refactoring and return constant pointers.
-      return const_cast<TfLiteRegistration*>(&registration_);
+      return &registration_;
     }
     return nullptr;
   }
-  TfLiteRegistration* FindOp(const char* op, int version) const override {
+  const TfLiteRegistration* FindOp(const char* op, int version) const override {
     return nullptr;
   }