Add int64 support for BroadcastArgs and BroadcastGradientArgs

[yongtang]

In array_ops.cc, both int32 and int64 are expected to be supported for BroadcastArgs and BroadcastGradientArgs. However, this was not the case as only int32 kernel are registered even though T is part of the TypeConstraint.

This fix adds the int64 kernel support for BroadcastArgs and BroadcastGradientArgs, and adds related test cases.

Signed-off-by: Yong Tang yong.tang.github@outlook.com

[drpngx]

Jenkins, test this please.

[vrv]

Thanks!

[drpngx]

Flaky test?

E1211 01:58:36.775779505      15 server_chttp2.cc:38]        {"created":"@1512957516.775699432","description":"No address added out of total 1 resolved","file":"external/grpc/src/core/ext/transport/chttp2/server/chttp2_server.cc","file_line":248,"referenced_errors":[{"created":"@1512957516.775696170","description":"Failed to add any wildcard listeners","file":"external/grpc/src/core/lib/iomgr/tcp_server_posix.cc","file_line":340,"referenced_errors":[{"created":"@1512957516.775683269","description":"Unable to configure socket","fd":13,"file":"external/grpc/src/core/lib/iomgr/tcp_server_utils_posix_common.cc","file_line":200,"referenced_errors":[{"created":"@1512957516.775666688","description":"OS Error","errno":98,"file":"external/grpc/src/core/lib/iomgr/tcp_server_utils_posix_common.cc","file_line":173,"os_error":"Address already in use","syscall":"bind"}]},{"created":"@1512957516.775695307","description":"Unable to configure socket","fd":13,"file":"external/grpc/src/core/lib/iomgr/tcp_server_utils_posix_common.cc","file_line":200,"referenced_errors":[{"created":"@1512957516.775691914","description":"OS Error","errno":98,"file":"external/grpc/src/core/lib/iomgr/tcp_server_utils_posix_common.cc","file_line":173,"os_error":"Address already in use","syscall":"bind"}]}]}]}
E.s...............
======================================================================
ERROR: testRemoteFunctionCrossProcess (__main__.FunctionalOpsTest)
----------------------------------------------------------------------
Traceback (most recent call last):
  File "/tmp/botexec/bazel-out/k8-py3-opt/bin/tensorflow/python/kernel_tests/functional_ops_test.runfiles/org_tensorflow/tensorflow/python/kernel_tests/functional_ops_test.py", line 588, in testRemoteFunctionCrossProcess
    workers, _ = test_util.create_local_cluster(2, 1)
  File "/tmp/botexec/bazel-out/k8-py3-opt/bin/tensorflow/python/kernel_tests/functional_ops_test.runfiles/org_tensorflow/tensorflow/python/framework/test_util.py", line 1298, in create_local_cluster
    for ix in range(num_ps)
  File "/tmp/botexec/bazel-out/k8-py3-opt/bin/tensorflow/python/kernel_tests/functional_ops_test.runfiles/org_tensorflow/tensorflow/python/framework/test_util.py", line 1298, in <listcomp>
    for ix in range(num_ps)
  File "/tmp/botexec/bazel-out/k8-py3-opt/bin/tensorflow/python/kernel_tests/functional_ops_test.runfiles/org_tensorflow/tensorflow/python/training/server_lib.py", line 145, in __init__
    self._server_def.SerializeToString(), status)
  File "/tmp/botexec/bazel-out/k8-py3-opt/bin/tensorflow/python/kernel_tests/functional_ops_test.runfiles/org_tensorflow/tensorflow/python/framework/errors_impl.py", line 473, in __exit__
    c_api.TF_GetCode(self.status.status))
tensorflow.python.framework.errors_impl.UnknownError: Could not start gRPC server

----------------------------------------------------------------------
Ran 39 tests in 2.226s

[drpngx]

Jenkins, test this please.

[drpngx]

Jenkins, test this please.