diff options
| author |  Yu-Cheng Ling <ycling@google.com> | 2018-06-01 16:27:45 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-01 16:30:28 -0700 |
| commit | b31498a054d55ce328a2820fd403af764c482500 (patch) | |
| tree | 91b8513149a36ae042e2a1b51f9e284701bbdcec /tensorflow/contrib/lite/toco/model.h | |
| parent | 73ec24e8b75ba4f73a06756502d8bf86b2a6828b (diff) | |
Support 5-inputs LSTM kernel in TFLite (float only).
PiperOrigin-RevId: 198943559
Diffstat (limited to 'tensorflow/contrib/lite/toco/model.h')
| -rw-r--r-- | tensorflow/contrib/lite/toco/model.h | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/tensorflow/contrib/lite/toco/model.h b/tensorflow/contrib/lite/toco/model.h index 9062c03c73..1a4f87e363 100644 --- a/tensorflow/contrib/lite/toco/model.h +++ b/tensorflow/contrib/lite/toco/model.h @@ -527,7 +527,15 @@ struct LstmCellOperator : Operator { ACTIV_TEMP = 3, NUM_OUTPUTS = 4 }; - LstmCellOperator() : Operator(OperatorType::kLstmCell) {} + enum KernelType { + KERNEL_BASIC = 0, + KERNEL_FULL = 1, + }; + + LstmCellOperator() + : Operator(OperatorType::kLstmCell), kernel_type(KERNEL_BASIC) {} + + KernelType kernel_type; }; // Element-wise multiplication operator. |