diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-05-07 12:37:36 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-07 16:58:51 -0700 |
commit | 6f3a890d91e6dbeb811aed23d0eb59abaa8c469f (patch) | |
tree | f05bad7a6c979613b43c495a042a675cbc59a13a /tensorflow/contrib/lite/nnapi_delegate.cc | |
parent | c3fef21c4ddf34fd68ab2cd44b0be497b5303b4e (diff) |
Adding Greater/GreaterEqual/LessEqual ops to complement Less.
PiperOrigin-RevId: 195704492
Diffstat (limited to 'tensorflow/contrib/lite/nnapi_delegate.cc')
-rw-r--r-- | tensorflow/contrib/lite/nnapi_delegate.cc | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/nnapi_delegate.cc b/tensorflow/contrib/lite/nnapi_delegate.cc index b4c46917bf..e903af87b7 100644 --- a/tensorflow/contrib/lite/nnapi_delegate.cc +++ b/tensorflow/contrib/lite/nnapi_delegate.cc @@ -372,7 +372,10 @@ void AddOpsAndParams(tflite::Interpreter* interpreter, case tflite::BuiltinOperator_MAXIMUM: case tflite::BuiltinOperator_MINIMUM: case tflite::BuiltinOperator_ARG_MAX: + case tflite::BuiltinOperator_GREATER: + case tflite::BuiltinOperator_GREATER_EQUAL: case tflite::BuiltinOperator_LESS: + case tflite::BuiltinOperator_LESS_EQUAL: case tflite::BuiltinOperator_NEG: FATAL("Op code %d is currently not delegated to NNAPI", builtin); nn_op_type = -1; // set to invalid |