tf.distribute.NcclAllReduce