diff options
author | 2018-05-07 15:41:22 -0700 | |
---|---|---|
committer | 2018-05-07 17:27:20 -0700 | |
commit | fc7f0b296dd53d1b72af21d36d36b6bcc5291ea7 (patch) | |
tree | 46e76ead2391a3fb1232459189ad0b8d0d8066ac /tensorflow/contrib/lite/nnapi_delegate.cc | |
parent | 3a2f1cfb73fa6a21eba077485bdc08aa05646ad1 (diff) |
Add support for select (via tf.where) support to tflite.
PiperOrigin-RevId: 195734246
Diffstat (limited to 'tensorflow/contrib/lite/nnapi_delegate.cc')
-rw-r--r-- | tensorflow/contrib/lite/nnapi_delegate.cc | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/nnapi_delegate.cc b/tensorflow/contrib/lite/nnapi_delegate.cc index e903af87b7..6a231dc6bc 100644 --- a/tensorflow/contrib/lite/nnapi_delegate.cc +++ b/tensorflow/contrib/lite/nnapi_delegate.cc @@ -377,6 +377,7 @@ void AddOpsAndParams(tflite::Interpreter* interpreter, case tflite::BuiltinOperator_LESS: case tflite::BuiltinOperator_LESS_EQUAL: case tflite::BuiltinOperator_NEG: + case tflite::BuiltinOperator_SELECT: FATAL("Op code %d is currently not delegated to NNAPI", builtin); nn_op_type = -1; // set to invalid break; |