Shape validation of `max_features` in `QuantizedReluX`
authorYong Tang <yong.tang.github@outlook.com>
Sun, 13 May 2018 14:26:06 +0000 (14:26 +0000)
committerYong Tang <yong.tang.github@outlook.com>
Mon, 14 May 2018 00:32:31 +0000 (00:32 +0000)
commit7c88788e63f3a747d2794175076db551d768734e
tree9b22939083e71a1a5aad5b5aa2f0cb597fdff472
parentf27033fb1212d7031a359c913d0f59e976b14c14
Shape validation of `max_features` in `QuantizedReluX`

In shape function of QuantizedReluX, `max_value` and
`min_features` have shape validation but not `max_features`.
This fix add restriction to `max_features` as well.

Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
tensorflow/core/ops/nn_ops.cc