From c3c83b2d465a91c535bd54269ba33a2442d7d36b Mon Sep 17 00:00:00 2001 From: ANSHUMAN TRIPATHY Date: Thu, 25 Apr 2019 15:45:19 +0530 Subject: [PATCH] [4] Review comments handled --- tensorflow/lite/kernels/concatenation.cc | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tensorflow/lite/kernels/concatenation.cc b/tensorflow/lite/kernels/concatenation.cc index 4e5a2d956d1..f17e955ba2e 100644 --- a/tensorflow/lite/kernels/concatenation.cc +++ b/tensorflow/lite/kernels/concatenation.cc @@ -129,7 +129,7 @@ TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { } \ } -#define TF_LITE_CONCATENATION_QUANTIZED \ +#define TF_LITE_CONCATENATION_QUANTIZED() \ { \ VectorOfQuantizedTensors all_inputs(*context, *node->inputs); \ tflite::ConcatenationParams op_params; \ @@ -158,7 +158,7 @@ TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { TF_LITE_CONCATENATION(int32); break; case kTfLiteUInt8: - TF_LITE_CONCATENATION_QUANTIZED; + TF_LITE_CONCATENATION_QUANTIZED(); break; case kTfLiteInt8: TF_LITE_CONCATENATION(int8_t); @@ -179,6 +179,8 @@ TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) { return kTfLiteOk; } +#undef TF_LITE_MACRO_DISPATCH + } // namespace concatenation TfLiteRegistration* Register_CONCATENATION_REF() {