Add support for time_major shape format to the sequential RNN Op in TF Lite.

This option, if set, changes the shape format of the inputs and outputs to
[max_time, batch_size, depth]. If false, it uses [batch_size, max_time, depth].
By default, it is set to false.

PiperOrigin-RevId: 182569507

1 files changed, 5 insertions, 0 deletions

diff --git a/tensorflow/contrib/lite/builtin_op_data.h b/tensorflow/contrib/lite/builtin_op_data.h
index 0c333f9e8c..3b43a1fd5d 100644
--- a/tensorflow/contrib/lite/builtin_op_data.h
+++ b/tensorflow/contrib/lite/builtin_op_data.h
@@ -83,6 +83,11 @@ typedef struct {
   TfLiteFusedActivation activation;
 } TfLiteRNNParams;
 
+typedef struct {
+  bool time_major;
+  TfLiteFusedActivation activation;
+} TfLiteSequenceRNNParams;
+
 typedef struct { TfLiteFusedActivation activation; } TfLiteFullyConnectedParams;
 
 typedef enum {