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>
Loading
Please sign in to comment