Enabling fp16 for NCCL 1 and 2.
authorA. Unique TensorFlower <gardener@tensorflow.org>
Mon, 9 Apr 2018 08:59:50 +0000 (01:59 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Mon, 9 Apr 2018 09:02:12 +0000 (02:02 -0700)
commit5c469e6bafb479ef110b2f02f070507a3711664d
tree57decbd46a5dc60c770c5a31e152b549e74e5758
parent1eea5ad3f9a622411117f7208d308055b0707d0f
Enabling fp16 for NCCL 1 and 2.

PiperOrigin-RevId: 192096789
tensorflow/contrib/nccl/kernels/nccl_manager.cc
tensorflow/contrib/nccl/kernels/nccl_manager_test.cc
tensorflow/contrib/nccl/ops/nccl_ops.cc
tensorflow/contrib/nccl/python/ops/nccl_ops_test.py