From 854ae599743a1e92a31ad49cfe42c6454cefd3b9 Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Tue, 9 Oct 2018 20:05:22 -0700
Subject: Use Ophints to support TfLite UnidirectionaSequenceLstm and add an
 e2e test.

Support peephole and num_proj as well.

PiperOrigin-RevId: 216467578
---
 tensorflow/contrib/lite/toco/model.h | 6 ++++++
 1 file changed, 6 insertions(+)

(limited to 'tensorflow/contrib/lite/toco/model.h')

diff --git a/tensorflow/contrib/lite/toco/model.h b/tensorflow/contrib/lite/toco/model.h
index 61f1f095e9..f3b84430db 100644
--- a/tensorflow/contrib/lite/toco/model.h
+++ b/tensorflow/contrib/lite/toco/model.h
@@ -58,6 +58,7 @@ enum class OperatorType : uint8 {
   kL2Normalization,
   kL2Pool,
   kLstmCell,
+  kUnidirectionalSequenceLstm,
   kLocalResponseNormalization,
   kLog,
   kLogistic,
@@ -635,6 +636,11 @@ struct LstmCellOperator : Operator {
   KernelType kernel_type;
 };
 
+struct UnidirectionalSequenceLstmOperator : Operator {
+  UnidirectionalSequenceLstmOperator()
+      : Operator(OperatorType::kUnidirectionalSequenceLstm) {}
+};
+
 // Element-wise multiplication operator.
 //
 // Inputs:
-- 
cgit v1.2.3