experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add gemmlowp label for SquaredDifference and Sum ops

Browse Source 
PiperOrigin-RevId: 258267779
This commit is contained in:
Karim Nosir 2019-07-15 17:07:19 -07:00 committed by TensorFlower Gardener
parent 0ea905b754
commit cdc1338193
2 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tensorflow/lite/kernels/reduce.cc
View File

@ -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 =

1
tensorflow/lite/kernels/squared_difference.cc
View File

@ -95,6 +95,7 @@ void EvalSquaredDifference(TfLiteContext* context, TfLiteNode* node,
TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) {
OpData* data = reinterpret_cast<OpData*>(node->user_data);
gemmlowp::ScopedProfilingLabel label("SquaredDifference");
const TfLiteTensor* input1 = GetInput(context, node, kInputTensor1);
const TfLiteTensor* input2 = GetInput(context, node, kInputTensor2);