Commit 69c74f1e authored by Yong Tang's avatar Yong Tang Committed by Rasmus Munk Larsen
Browse files

Add int64 support for output_shape of tf.nn.conv3d_transpose (#19248)



* Add int64 support for output_shape of tf.nn.conv3d_transpose

This fix tries to address the issue raised in 18887 where
the output_shape of tf.nn.conv3d_transpose only support
int32 data types. The support of int64 has been added in this PR
with test case covered.

This fix fixes 18887.

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

* Update op registration for Conv3DBackpropInputV2

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

* Add test case for int64 support of output_shape with tf.nn.conv3d_transpose

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

* Update test case with both int32 and int64

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

* Fix pylint issue

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