From 3b458633e15e816ecd10cabab934fa16b9200534 Mon Sep 17 00:00:00 2001 From: Koan-Sin Tan Date: Fri, 6 Mar 2020 17:04:43 +0800 Subject: [PATCH] [tflite] NNAPI Pow op allows scalar input --- tensorflow/lite/delegates/nnapi/nnapi_delegate.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/tensorflow/lite/delegates/nnapi/nnapi_delegate.cc b/tensorflow/lite/delegates/nnapi/nnapi_delegate.cc index a90b28fcd10..760c24c63f5 100644 --- 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;