Commit f2e1ac3e authored by Yong Tang's avatar Yong Tang
Browse files

Adjust _NHWCToNCHW and _NCHWToNHWC in bias_op_test.py



The _NHWCToNCHW and _NCHWToNHWC in bias_op_test.py
only took into considerations about 4D situation,
as it assume that third-to-last dim is the channel
dimension.

This fix adjust to use second-to-first dim as the channel
dimension so that 3D/4D/5D data could be handled by NCHW format.

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>
parent 08a8a721
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment