diff options
author | 2018-08-07 10:18:12 -0700 | |
---|---|---|
committer | 2018-08-07 10:23:42 -0700 | |
commit | 1a22e9a608b43cbdf7d990d09c7d317ee3c57d8c (patch) | |
tree | a6cc25e02614e6886a7289082634b56db413259f /tensorflow/contrib/lite/nnapi_delegate.cc | |
parent | 36a606ea383596ce687c5262466ae342eaf7d412 (diff) |
Add NNAPI delegation support for MEAN, RNN, SVDF, LSTM and lookup functions.
PiperOrigin-RevId: 207736388
Diffstat (limited to 'tensorflow/contrib/lite/nnapi_delegate.cc')
-rw-r--r-- | tensorflow/contrib/lite/nnapi_delegate.cc | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/tensorflow/contrib/lite/nnapi_delegate.cc b/tensorflow/contrib/lite/nnapi_delegate.cc index c91f488175..13325a8c7c 100644 --- a/tensorflow/contrib/lite/nnapi_delegate.cc +++ b/tensorflow/contrib/lite/nnapi_delegate.cc @@ -568,9 +568,17 @@ TfLiteStatus AddOpsAndParams( "NNAPI does not support L2Normalization with fused activations"); } break; + case tflite::BuiltinOperator_HASHTABLE_LOOKUP: + if (interpreter->tensor(node.outputs->data[0])->type != + kTfLiteFloat32) { + logError("NNAPI only support HASHTABLE_LOOKUP with float32 output", + builtin); + return kTfLiteError; + } + nn_op_type = ANEURALNETWORKS_HASHTABLE_LOOKUP; + break; case tflite::BuiltinOperator_CONCAT_EMBEDDINGS: case tflite::BuiltinOperator_LSH_PROJECTION: - case tflite::BuiltinOperator_HASHTABLE_LOOKUP: case tflite::BuiltinOperator_BIDIRECTIONAL_SEQUENCE_RNN: case tflite::BuiltinOperator_UNIDIRECTIONAL_SEQUENCE_RNN: case tflite::BuiltinOperator_EMBEDDING_LOOKUP_SPARSE: |