Add axis support for `tf.nn.crelu` (#15836)
* Add axis support for tf.nn.crelu
This fix tries to address the issue raised in 15619
where it was not possible to specify an `axis` for
`tf.nn.crelu`. By default, `axis=-1` was used for concatenation
implicitly.
This fix adds the support of `axis` for tf.nn.crelu,
and adds test cases for it.
This fix fixes 15619.
Signed-off-by:
Yong Tang <yong.tang.github@outlook.com>
Loading
Please sign in to comment