diff --git a/tensorflow/lite/kernels/reduce.cc b/tensorflow/lite/kernels/reduce.cc index c5e2c673497..d28ec70f98a 100644 --- a/tensorflow/lite/kernels/reduce.cc +++ b/tensorflow/lite/kernels/reduce.cc @@ -523,6 +523,7 @@ TfLiteStatus EvalGeneric(TfLiteContext* context, TfLiteNode* node) { TfLiteStatus EvalSum(TfLiteContext* context, TfLiteNode* node) { OpContext op_context(context, node); + gemmlowp::ScopedProfilingLabel label("Sum"); const auto& input = op_context.input; const auto& output = op_context.output; const bool same_scale = diff --git a/tensorflow/lite/kernels/squared_difference.cc b/tensorflow/lite/kernels/squared_difference.cc index 3661cf9f98c..3ec9838893a 100644 --- a/tensorflow/lite/kernels/squared_difference.cc +++ b/tensorflow/lite/kernels/squared_difference.cc @@ -95,6 +95,7 @@ void EvalSquaredDifference(TfLiteContext* context, TfLiteNode* node, TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { OpData* data = reinterpret_cast(node->user_data); + gemmlowp::ScopedProfilingLabel label("SquaredDifference"); const TfLiteTensor* input1 = GetInput(context, node, kInputTensor1); const TfLiteTensor* input2 = GetInput(context, node, kInputTensor2);