diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-07-18 14:12:13 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-18 14:15:24 -0700 |
commit | 4ca04537c0d1d75ea37944aa3bb2dc749428031a (patch) | |
tree | 663b7a6b36e560258dd95a603e0440a4c7b278ac /tensorflow/contrib/lite/toco/graph_transformations/propagate_array_data_types.cc | |
parent | 44af531d952a35c887770ecc4cfddfb0431c2478 (diff) |
Import/export support for Any, LogicalAnd, and LogicalNot ops.
PiperOrigin-RevId: 205134621
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, 3 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 670bcf64e7..3dda536ef7 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 @@ -62,6 +62,9 @@ bool PropagateArrayDataTypes::Run(Model* model, std::size_t op_index) { case OperatorType::kGreaterEqual: case OperatorType::kEqual: case OperatorType::kNotEqual: + case OperatorType::kAny: + case OperatorType::kLogicalAnd: + case OperatorType::kLogicalNot: // These operators unconditionally produce bool outputs SetDataTypeForAllOutputs(model, op, ArrayDataType::kBool); break; |