Fix a test.

- SetCustomOp also sets the name of the custom op. Test was checking against
the wrong name in the profile.

PiperOrigin-RevId: 215665359

1 files changed, 4 insertions, 2 deletions

diff --git a/tensorflow/contrib/lite/profiling/profile_summarizer_test.cc b/tensorflow/contrib/lite/profiling/profile_summarizer_test.cc
index 67a5eecfa0..465c294962 100644
--- a/tensorflow/contrib/lite/profiling/profile_summarizer_test.cc
+++ b/tensorflow/contrib/lite/profiling/profile_summarizer_test.cc
@@ -31,6 +31,8 @@ namespace profiling {
 
 namespace {
 
+const char* kOpName = "SimpleOpEval";
+
 #ifdef TFLITE_PROFILING_ENABLED
 TfLiteStatus SimpleOpEval(TfLiteContext* context, TfLiteNode* node) {
   const TfLiteTensor* input1 = tflite::GetInput(context, node, /*index=*/0);
@@ -63,7 +65,7 @@ TfLiteRegistration* RegisterSimpleOpWithProfilingDetails() {
                                             SimpleOpEval,
                                             SimpleOpProfilingString,
                                             tflite::BuiltinOperator_CUSTOM,
-                                            "SimpleOpEval",
+                                            kOpName,
                                             1};
   return &registration;
 }
@@ -89,7 +91,7 @@ void SimpleOpModel::Init(
   inputs_[0] = AddInput({TensorType_INT32, {1}});
   inputs_[1] = AddInput({TensorType_INT32, {1}});
   output_ = AddOutput({TensorType_INT32, {}});
-  SetCustomOp("SimpleAdd", {}, registration);
+  SetCustomOp(kOpName, {}, registration);
   BuildInterpreter({GetShape(inputs_[0]), GetShape(inputs_[1])});
 }