Commit 4a7bbb54 authored by Yong Tang's avatar Yong Tang Committed by Shanqing Cai
Browse files

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: default avatarYong Tang <yong.tang.github@outlook.com>
parent f5de234d
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment