Merge pull request #37383 from freedomtan:pow_allow_scalar_input

PiperOrigin-RevId: 301437812 Change-Id: Id02d8bc3f0843f33c1cdd4f90429ec338e0143ee
2020-03-17 13:17:34 -07:00 · 2020-03-17 13:17:34 -07:00 · ab407b3fbe
commit ab407b3fbe
parent 1a101d9163 3b458633e1
1 changed files with 1 additions and 0 deletions
--- a/tensorflow/lite/delegates/nnapi/nnapi_delegate.cc
+++ b/tensorflow/lite/delegates/nnapi/nnapi_delegate.cc
@ -149,6 +149,7 @@ bool IsScalarInputSupported(int builtin_code) {
    case kTfLiteBuiltinGreaterEqual:
    case kTfLiteBuiltinLess:
    case kTfLiteBuiltinLessEqual:
+    case kTfLiteBuiltinPow:
      return true;
    default:
      return false;