Add int64 support for output_shape of tf.nn.conv3d_transpose (#19248)
authorYong Tang <yong.tang.github@outlook.com>
Mon, 14 May 2018 18:35:40 +0000 (11:35 -0700)
committerRasmus Munk Larsen <rmlarsen@google.com>
Mon, 14 May 2018 18:35:40 +0000 (11:35 -0700)
commit69c74f1e74eb5da964638533d594475ee9e54a66
tree1f8e263033d86d178024c4cda05e4a7c9700522e
parentc0dd7852bfa216e0c9bc9eeb57d2e613f7996116
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>
tensorflow/core/kernels/conv_grad_ops_3d.cc
tensorflow/core/ops/nn_ops.cc
tensorflow/python/kernel_tests/conv3d_transpose_test.py