diff options
author | 2018-06-19 09:49:17 -0700 | |
---|---|---|
committer | 2018-06-19 09:54:05 -0700 | |
commit | 5fc2bdd2d5f624a6bad9e83b992029e3799ab64e (patch) | |
tree | 576404472bd26801fe2fce6f6385669aef1f42e4 /tensorflow/contrib/lite/nnapi_delegate.cc | |
parent | c532c3f319c72074e6fb8cb10c6d05a3839bcc0a (diff) |
Implement TFLite sqrt/rsqrt unary operators
PiperOrigin-RevId: 201191877
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 8d506f562f..1e012c89ae 100644 --- a/tensorflow/contrib/lite/nnapi_delegate.cc +++ b/tensorflow/contrib/lite/nnapi_delegate.cc @@ -501,6 +501,8 @@ void AddOpsAndParams(tflite::Interpreter* interpreter, case tflite::BuiltinOperator_EQUAL: case tflite::BuiltinOperator_NOT_EQUAL: case tflite::BuiltinOperator_SUM: + case tflite::BuiltinOperator_SQRT: + case tflite::BuiltinOperator_RSQRT: FATAL("Op code %d is currently not delegated to NNAPI", builtin); nn_op_type = -1; // set to invalid break; |