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

Check CONCAT input type for NNAPI delegate

Browse Source 
PiperOrigin-RevId: 357741790
Change-Id: I753873a958f36b89fb56fc7284bf4247e2ead68a
This commit is contained in:
Sachin Joglekar 2021-02-16 09:54:37 -08:00 committed by TensorFlower Gardener
parent 6de1774b27
commit b2e33765a1
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/lite/delegates/nnapi/nnapi_delegate.cc
View File

@ -2057,7 +2057,11 @@ bool NNAPIDelegateKernel::Validate(
NNAPIValidationFailureType::kUnsupportedOperandRank,
"Input rank should be less than 4", &val_ctx);
if (context->tensors[node->inputs->data[0]].type == kTfLiteUInt8 &&
const auto& input_type = context->tensors[node->inputs->data[0]].type;
EXPECT_INPUT_TYPE_IN(input_type, kTfLiteFloat16, kTfLiteFloat32,
kTfLiteUInt8, kTfLiteInt8);
if (input_type == kTfLiteUInt8 &&
android_sdk_version < kMinSdkVersionForNNAPI12) {
auto first_param = context->tensors[node->inputs->data[0]].params;
for (int i = 1; i < node->inputs->size; i++) {