Fix BiasAddGradShape for NCHW, channel should be second from the left,
not third-from-right as we could have 3/4/5 dimensional cases
Signed-off-by:
Yong Tang <yong.tang.github@outlook.com>
Loading
Please sign in to comment
not third-from-right as we could have 3/4/5 dimensional cases
Signed-off-by:
Yong Tang <yong.tang.github@outlook.com>