Commit 5baa4399 authored by Yong Tang's avatar Yong Tang Committed by Rasmus Munk Larsen
Browse files

Fix issue caused by None in batch dimension for tf.layers.conv3d (#18027)



* Fix issue caused by None in batch dimension for tf.layers.conv3d

This fix tries to address the issue raised in 15655 where
error returns when the batch dimension for tf.layers.conv3d
is None with "channels_first" format.

This fix cast None to `-1` to address the issue

This fix fixes 15655.

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>

* Add test case for batch dimension = None in tf.layers.conv3d

Signed-off-by: default avatarYong Tang <yong.tang.github@outlook.com>

* Address review feedback

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