diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-06-07 02:05:06 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-07 02:07:45 -0700 |
| commit | c70b7128bfb9f0283c60bbec8fd7b0c12f741d95 (patch) | |
| tree | 49a75161cb036b87817436d2bad9b79bfbb61425 /tensorflow/contrib/lite/toco/graph_transformations/propagate_array_data_types.cc | |
| parent | c2368f875b53e9144a1803a3e67c5a61aa9c5862 (diff) | |
Implementation of TensorFlowEqual and TensorFlowNotEqual.
PiperOrigin-RevId: 199602232
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 | 2 |
1 files changed, 2 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 64096fb069..92d283ca2c 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 @@ -60,6 +60,8 @@ bool PropagateArrayDataTypes::Run(Model* model, std::size_t op_index) { case OperatorType::kTensorFlowLessEqual: case OperatorType::kTensorFlowGreater: case OperatorType::kTensorFlowGreaterEqual: + case OperatorType::kTensorFlowEqual: + case OperatorType::kTensorFlowNotEqual: // These operators unconditionally produce bool outputs SetDataTypeForAllOutputs(model, op, ArrayDataType::kBool); break; |