experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add LSTM to op spec. This CL only adds the intermediates fields for one variants of LSTM.

Browse Source 
PiperOrigin-RevId: 278703956
Change-Id: I4468e546d4a1096b7345d4e96701fec7d4f8ec59
This commit is contained in:
Jian Li 2019-11-05 14:04:23 -08:00 committed by TensorFlower Gardener
parent 78f7b4165e
commit 60b7d06456
1 changed files with 9 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tensorflow/lite/tools/optimize/operator_property.cc
View File

@ -165,6 +165,15 @@ OperatorProperty GetOperatorProperty(const ModelT* model, int subgraph_index,
property.version = 2;
break;
}
case BuiltinOperator_LSTM: {
// TODO(jianlijianli): extend LSTM op spec to inlucde input, bias etc.
// TODO(jianlijianli): extend this to other variants of LSTM.
// LSTM need 5 intermediate tensors. This agrees with the fully quantized
// kernels in lstm_eval.cc
property.intermediates = {{0, {}}, {1, {}}, {2, {}}, {3, {}}, {4, {}}};
property.version = 2;
break;
}
case BuiltinOperator_L2_NORMALIZATION: {
property.inputs = {{0, {}}};
// L2 Norm requires output with 1/128 as scale and 0 as zero point.