diff options
author | 2018-08-14 07:35:06 -0700 | |
---|---|---|
committer | 2018-08-14 07:39:12 -0700 | |
commit | 62db4ac8780f37db2a9c5d18faa755e58e6a0403 (patch) | |
tree | b867a85562f7e80cbd763f7cb65d6ce74dcd6b93 /tensorflow/contrib/lite/toco/graph_transformations/resolve_constant_unary.cc | |
parent | 9b05ad3798e011202c8a0f094a6507855a0a9f2d (diff) |
Fixing bad conditional.
PiperOrigin-RevId: 208644378
Diffstat (limited to 'tensorflow/contrib/lite/toco/graph_transformations/resolve_constant_unary.cc')
-rw-r--r-- | tensorflow/contrib/lite/toco/graph_transformations/resolve_constant_unary.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/lite/toco/graph_transformations/resolve_constant_unary.cc b/tensorflow/contrib/lite/toco/graph_transformations/resolve_constant_unary.cc index fe3882c28d..475415e481 100644 --- a/tensorflow/contrib/lite/toco/graph_transformations/resolve_constant_unary.cc +++ b/tensorflow/contrib/lite/toco/graph_transformations/resolve_constant_unary.cc @@ -246,8 +246,8 @@ bool ResolveConstantUnaryOperator::Run(Model* model, std::size_t op_index) { } output_float_data[i] = outval; } - } else if (unary_op->type == OperatorType::kRelu6 && - unary_op->type == OperatorType::kRelu1 && + } else if (unary_op->type == OperatorType::kRelu6 || + unary_op->type == OperatorType::kRelu1 || unary_op->type == OperatorType::kRelu) { for (size_t i = 0; i < output_buffer_size; ++i) { const float value = (*input_float_data)[i]; |