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

Merge pull request #37383 from freedomtan:pow_allow_scalar_input

Browse Source 
PiperOrigin-RevId: 301437812
Change-Id: Id02d8bc3f0843f33c1cdd4f90429ec338e0143ee
This commit is contained in:
TensorFlower Gardener 2020-03-17 13:17:34 -07:00
commit ab407b3fbe
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -149,6 +149,7 @@ bool IsScalarInputSupported(int builtin_code) {
case kTfLiteBuiltinGreaterEqual: case kTfLiteBuiltinGreaterEqual:
case kTfLiteBuiltinLess: case kTfLiteBuiltinLess:
case kTfLiteBuiltinLessEqual: case kTfLiteBuiltinLessEqual:
case kTfLiteBuiltinPow:
return true; return true;
default: default:
return false; return false;