diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-07-02 15:13:59 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-02 15:17:08 -0700 |
| commit | 990677c71d07dace05f50afda918797af37c5fa1 (patch) | |
| tree | d59dc0fab07ae689c12c67a5874afb20eaf2fd71 /tensorflow/contrib/lite/nnapi_delegate.cc | |
| parent | 7a66008cf6f07c9ad7803a3dd9ac33ddd46b1dcf (diff) | |
Fix NNAPI delegation for Sub and Div.
PiperOrigin-RevId: 203020841
Diffstat (limited to 'tensorflow/contrib/lite/nnapi_delegate.cc')
| -rw-r--r-- | tensorflow/contrib/lite/nnapi_delegate.cc | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/nnapi_delegate.cc b/tensorflow/contrib/lite/nnapi_delegate.cc index 44cef80ac3..905c0919cb 100644 --- a/tensorflow/contrib/lite/nnapi_delegate.cc +++ b/tensorflow/contrib/lite/nnapi_delegate.cc @@ -534,10 +534,14 @@ TfLiteStatus AddOpsAndParams( case tflite::BuiltinOperator_DIV: nnapi_version = 11; // require NNAPI 1.1 nn_op_type = ANEURALNETWORKS_DIV; + check_and_add_activation( + reinterpret_cast<TfLiteDivParams*>(node.builtin_data)->activation); break; case tflite::BuiltinOperator_SUB: nnapi_version = 11; // require NNAPI 1.1 nn_op_type = ANEURALNETWORKS_SUB; + check_and_add_activation( + reinterpret_cast<TfLiteSubParams*>(node.builtin_data)->activation); break; case tflite::BuiltinOperator_SQUEEZE: nnapi_version = 11; // requires NNAPI 1.1 |