From 990677c71d07dace05f50afda918797af37c5fa1 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Mon, 2 Jul 2018 15:13:59 -0700 Subject: Fix NNAPI delegation for Sub and Div. PiperOrigin-RevId: 203020841 --- tensorflow/contrib/lite/nnapi_delegate.cc | 4 ++++ 1 file changed, 4 insertions(+) (limited to 'tensorflow/contrib/lite/nnapi_delegate.cc') 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(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(node.builtin_data)->activation); break; case tflite::BuiltinOperator_SQUEEZE: nnapi_version = 11; // requires NNAPI 1.1 -- cgit v1.2.3