Searched refs:mpi_collectives (Results 1 – 15 of 15) sorted by relevance
128 from tensorflow.contrib.mpi_collectives.python.ops.mpi_ops import init129 from tensorflow.contrib.mpi_collectives.python.ops.mpi_ops import size130 from tensorflow.contrib.mpi_collectives.python.ops.mpi_ops import rank131 from tensorflow.contrib.mpi_collectives.python.ops.mpi_ops import local_rank132 from tensorflow.contrib.mpi_collectives.python.ops.mpi_ops import allgather133 from tensorflow.contrib.mpi_collectives.python.ops.mpi_ops import _allreduce
24 import tensorflow.contrib.mpi_collectives as mpi
18 package tensorflow.contrib.mpi_collectives;
23 import tensorflow.contrib.mpi_collectives as mpi
27 import tensorflow.contrib.mpi_collectives as mpi
12 tensorflow/contrib/mpi_collectives
310 tensorflow/contrib/mpi_collectives/python311 tensorflow/contrib/mpi_collectives/python/ops
23 namespace mpi_collectives { namespace
24 namespace mpi_collectives { namespace
26 namespace mpi_collectives { namespace
40 namespace mpi_collectives {
83 namespace mpi_collectives { namespace
23 from tensorflow.contrib.mpi_collectives.ops import gen_mpi_ops
114 ] + if_mpi(["//tensorflow/contrib/mpi_collectives:mpi_collectives_py"]) + select({181 ] + if_mpi(["//tensorflow/contrib/mpi_collectives:mpi_collectives_py"]) + select({
226 tensorflow/third_party/mpi_collectives/BUILD