diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-07-16 19:09:52 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-16 19:13:36 -0700 |
| commit | 1a2af489b1087eb22ec76863867e4e397e453e34 (patch) | |
| tree | d2576dbbfc6f77897e59353e0844aa0c75a23033 /tensorflow/contrib/lite/nnapi_delegate.cc | |
| parent | d8f3425e5b054dff01b5ece80e8c8a101c4ed816 (diff) | |
Support reduce_max and reduce_prod
PiperOrigin-RevId: 204846139
Diffstat (limited to 'tensorflow/contrib/lite/nnapi_delegate.cc')
| -rw-r--r-- | tensorflow/contrib/lite/nnapi_delegate.cc | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/nnapi_delegate.cc b/tensorflow/contrib/lite/nnapi_delegate.cc index cc668485a4..5950840e8a 100644 --- a/tensorflow/contrib/lite/nnapi_delegate.cc +++ b/tensorflow/contrib/lite/nnapi_delegate.cc @@ -607,6 +607,8 @@ TfLiteStatus AddOpsAndParams( case tflite::BuiltinOperator_EQUAL: case tflite::BuiltinOperator_NOT_EQUAL: case tflite::BuiltinOperator_SUM: + case tflite::BuiltinOperator_REDUCE_MAX: + case tflite::BuiltinOperator_REDUCE_PROD: case tflite::BuiltinOperator_SQRT: case tflite::BuiltinOperator_RSQRT: case tflite::BuiltinOperator_SHAPE: |