* Added code to delegate LSTM operation to NN API. Off by default.

* Added code to turn on NN API delegation for speaker id test. Off by default.

PiperOrigin-RevId: 181666821

1 files changed, 8 insertions, 1 deletions

diff --git a/tensorflow/contrib/lite/models/speech_speakerid_model_test.cc b/tensorflow/contrib/lite/models/speech_speakerid_model_test.cc
index 9da0fb1fc6..e208fac8df 100644
--- a/tensorflow/contrib/lite/models/speech_speakerid_model_test.cc
+++ b/tensorflow/contrib/lite/models/speech_speakerid_model_test.cc
@@ -43,7 +43,7 @@ constexpr int kLstmLayer3OutputStateTensor = 61;
 constexpr int kLstmLayer3CellStateTensor = 62;
 constexpr int kModelOutputTensor = 66;
 
-TEST(SpeechSpeakerId, OkGoogleTest) {
+void SpeakerIdTest(bool useNNAPI) {
   // Read the model.
   string tflite_file_path =
       StrCat(TestDataPath(), "/", "speech_speakerid_model.tflite");
@@ -56,6 +56,9 @@ TEST(SpeechSpeakerId, OkGoogleTest) {
   std::unique_ptr<Interpreter> interpreter;
   InterpreterBuilder(*model, resolver)(&interpreter);
   CHECK(interpreter != nullptr);
+
+  interpreter->UseNNAPI(useNNAPI);
+
   interpreter->AllocateTensors();
 
   // Load the input frames.
@@ -110,5 +113,9 @@ TEST(SpeechSpeakerId, OkGoogleTest) {
   }
 }
 
+TEST(SpeechSpeakerId, OkGoogleTest) { SpeakerIdTest(false); }
+
+TEST(SpeechSpeakerId, OkGoogleTestUsingNNAPI) { SpeakerIdTest(true); }
+
 }  // namespace models
 }  // namespace tflite