diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-10-09 20:05:22 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-09 20:08:54 -0700 |
commit | 854ae599743a1e92a31ad49cfe42c6454cefd3b9 (patch) | |
tree | 1ff75695f61c5eb3353e739295e81f76bbe28a64 /tensorflow/contrib/lite/toco/graph_transformations/propagate_array_data_types.cc | |
parent | 58fcfc98cd59ae3952399fc55380b8733df08df9 (diff) |
Use Ophints to support TfLite UnidirectionaSequenceLstm and add an e2e test.
Support peephole and num_proj as well.
PiperOrigin-RevId: 216467578
Diffstat (limited to 'tensorflow/contrib/lite/toco/graph_transformations/propagate_array_data_types.cc')
-rw-r--r-- | tensorflow/contrib/lite/toco/graph_transformations/propagate_array_data_types.cc | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/toco/graph_transformations/propagate_array_data_types.cc b/tensorflow/contrib/lite/toco/graph_transformations/propagate_array_data_types.cc index 40cd6dea82..47faa20a29 100644 --- a/tensorflow/contrib/lite/toco/graph_transformations/propagate_array_data_types.cc +++ b/tensorflow/contrib/lite/toco/graph_transformations/propagate_array_data_types.cc @@ -239,6 +239,12 @@ void SetDataTypeForAllOutputs(Model* model, Operator* op, } break; } + case OperatorType::kUnidirectionalSequenceLstm: { + const ArrayDataType data_type = model->GetArray(op->inputs[0]).data_type; + if (data_type != ArrayDataType::kFloat) return ::tensorflow::Status::OK(); + SetDataTypeForAllOutputs(model, op, data_type); + break; + } default: { // These operators produce outputs with the same type as their 1st input CHECK_GT(op->inputs.size(), 0); |