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

[4] Review comments handled

Browse Source
This commit is contained in:
ANSHUMAN TRIPATHY 2019-04-25 15:45:19 +05:30
parent 79122e21c8
commit c3c83b2d46
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/lite/kernels/concatenation.cc
View File

@ -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); \ VectorOfQuantizedTensors all_inputs(*context, *node->inputs); \
tflite::ConcatenationParams op_params; \ tflite::ConcatenationParams op_params; \
@ -158,7 +158,7 @@ TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) {
TF_LITE_CONCATENATION(int32); TF_LITE_CONCATENATION(int32);
break; break;
case kTfLiteUInt8: case kTfLiteUInt8:
TF_LITE_CONCATENATION_QUANTIZED; TF_LITE_CONCATENATION_QUANTIZED();
break; break;
case kTfLiteInt8: case kTfLiteInt8:
TF_LITE_CONCATENATION(int8_t); TF_LITE_CONCATENATION(int8_t);
@ -179,6 +179,8 @@ TfLiteStatus Eval(TfLiteContext* context, TfLiteNode* node) {
return kTfLiteOk; return kTfLiteOk;
} }
#undef TF_LITE_MACRO_DISPATCH
} // namespace concatenation } // namespace concatenation
TfLiteRegistration* Register_CONCATENATION_REF() { TfLiteRegistration* Register_CONCATENATION_REF() {