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

Merge pull request #27506 from ANSHUMAN87:comparison-bug-fix

Browse Source 
PiperOrigin-RevId: 242246604
This commit is contained in:
TensorFlower Gardener 2019-04-05 22:56:19 -07:00
commit f0cccb0705
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/lite/kernels/comparisons.cc
View File

@ -37,9 +37,8 @@ TfLiteStatus ComparisonPrepare(TfLiteContext* context, TfLiteNode* node) {
const TfLiteTensor* input2 = GetInput(context, node, kInputTensor2); const TfLiteTensor* input2 = GetInput(context, node, kInputTensor2);
TfLiteTensor* output = GetOutput(context, node, kOutputTensor); TfLiteTensor* output = GetOutput(context, node, kOutputTensor);
// Don't support string and bool. // Don't support string.
TF_LITE_ENSURE(context, TF_LITE_ENSURE(context, input1->type != kTfLiteString);
input1->type != kTfLiteString || input1->type != kTfLiteBool);
// Currently only support tensors have the same type. // Currently only support tensors have the same type.
TF_LITE_ENSURE_TYPES_EQ(context, input1->type, input2->type); TF_LITE_ENSURE_TYPES_EQ(context, input1->type, input2->type);
output->type = kTfLiteBool; output->type = kTfLiteBool;