diff options
author | 2018-06-01 16:27:45 -0700 | |
---|---|---|
committer | 2018-06-01 16:30:28 -0700 | |
commit | b31498a054d55ce328a2820fd403af764c482500 (patch) | |
tree | 91b8513149a36ae042e2a1b51f9e284701bbdcec /tensorflow/contrib/lite/builtin_op_data.h | |
parent | 73ec24e8b75ba4f73a06756502d8bf86b2a6828b (diff) |
Support 5-inputs LSTM kernel in TFLite (float only).
PiperOrigin-RevId: 198943559
Diffstat (limited to 'tensorflow/contrib/lite/builtin_op_data.h')
-rw-r--r-- | tensorflow/contrib/lite/builtin_op_data.h | 10 |
1 files changed, 10 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/builtin_op_data.h b/tensorflow/contrib/lite/builtin_op_data.h index 52ab9ee640..c1cc4476fb 100644 --- a/tensorflow/contrib/lite/builtin_op_data.h +++ b/tensorflow/contrib/lite/builtin_op_data.h @@ -148,10 +148,20 @@ typedef struct { float beta; } TfLiteLocalResponseNormParams; +typedef enum { + kTfLiteLSTMFullKernel = 0, + kTfLiteLSTMBasicKernel +} TfLiteLSTMKernelType; + typedef struct { + // Parameters for LSTM version 1. TfLiteFusedActivation activation; float cell_clip; float proj_clip; + + // Parameters for LSTM version 2. + // kTfLiteLSTMBasicKernel is only supported in version 2 or above. + TfLiteLSTMKernelType kernel_type; } TfLiteLSTMParams; typedef struct { |