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: Yong Tang <yong.tang.github@outlook.com>
* Update op registration for Conv3DBackpropInputV2
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Add test case for int64 support of output_shape with tf.nn.conv3d_transpose
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Update test case with both int32 and int64
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
* Fix pylint issue
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>