From 37f3582c7f322ef3b78ea019dfca63382671ef3d Mon Sep 17 00:00:00 2001 From: =?utf8?q?PRAVEEN=20DORESWAMY=20NAIDU/System=20SW=20/SRI-Bangalore/St?= =?utf8?q?aff=20Engineer/=EC=82=BC=EC=84=B1=EC=A0=84=EC=9E=90?= Date: Mon, 19 Nov 2018 13:51:52 +0530 Subject: [PATCH] Nnapi delegate for ArgMax (#3558) The changes introduces the nnapi delegate for ArgMax operation. Signed-off-by: Praveen D N --- libs/support/tflite/src/nnapi_delegate.cpp | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/libs/support/tflite/src/nnapi_delegate.cpp b/libs/support/tflite/src/nnapi_delegate.cpp index 294997e..cfeba89 100644 --- a/libs/support/tflite/src/nnapi_delegate.cpp +++ b/libs/support/tflite/src/nnapi_delegate.cpp @@ -721,6 +721,13 @@ TfLiteStatus AddOpsAndParams( augmented_inputs.data(), static_cast(node.outputs->size), reinterpret_cast(node.outputs->data))); continue; + case tflite::BuiltinOperator_ARG_MAX: + CHECK_NN(ANeuralNetworksModel_addOperationEx( + nn_model, ANEURALNETWORKS_ARGMAX_EX, + static_cast(augmented_inputs.size()), + augmented_inputs.data(), static_cast(node.outputs->size), + reinterpret_cast(node.outputs->data))); + continue; case tflite::BuiltinOperator_CONCAT_EMBEDDINGS: case tflite::BuiltinOperator_LSH_PROJECTION: case tflite::BuiltinOperator_BIDIRECTIONAL_SEQUENCE_RNN: @@ -748,7 +755,6 @@ TfLiteStatus AddOpsAndParams( case tflite::BuiltinOperator_PRELU: case tflite::BuiltinOperator_MAXIMUM: case tflite::BuiltinOperator_MINIMUM: - case tflite::BuiltinOperator_ARG_MAX: #ifndef OBS_BUILD case tflite::BuiltinOperator_ARG_MIN: #endif -- 2.7.4