diff options
| author |  Zhixian Yan <zhixianyan@google.com> | 2018-01-19 14:51:11 -0800 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-01-19 14:54:54 -0800 |
| commit | 0c4c353690b0cec2e26e10abd59f66f7e61cc974 (patch) | |
| tree | 1721bee3f147495994fa73a5b75ba1bc554e884e /tensorflow/contrib/lite/toco/graph_transformations/propagate_array_data_types.cc | |
| parent | 87fb310d4cc6cea081ece27c621104dd6901ba33 (diff) | |
TFLite export optional tensor as -1 id.
PiperOrigin-RevId: 182592943
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 | 3 |
1 files changed, 2 insertions, 1 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 c6f17cf319..29b55d9bfc 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 @@ -38,7 +38,8 @@ bool PropagateArrayDataTypes::Run(Model* model, std::size_t op_index) { // If the data type of some input is unknown, we need to yield. for (const auto& input : op->inputs) { - if (model->arrays[input]->data_type == ArrayDataType::kNone) { + if (!model->IsOptionalArray(input) && + model->arrays[input]->data_type == ArrayDataType::kNone) { return false; } } |