| /external/tensorflow/tensorflow/contrib/all_reduce/ |
| D | BUILD | 17 ":all_reduce",
23 name = "all_reduce",
25 "python/all_reduce.py",
30 "//tensorflow/python/distribute:all_reduce",
|
| D | __init__.py | 22 from tensorflow.contrib.all_reduce.python.all_reduce import *
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | ar_crs_combiner.cc | 324 auto all_reduce = pair.ar; in RewriteGraph() local
325 auto parent_computation = all_reduce->parent(); in RewriteGraph()
326 auto all_reduce_id = all_reduce->all_reduce_id(); in RewriteGraph()
327 auto prev = all_reduce->mutable_operand(0); in RewriteGraph()
328 auto next = all_reduce->users()[0]; in RewriteGraph()
329 TF_CHECK_OK(all_reduce->ReplaceUseWith(next, prev)); in RewriteGraph()
330 TF_CHECK_OK(parent_computation->RemoveInstruction(all_reduce)); in RewriteGraph()
|
| D | ar_crs_combiner.h | 91 ArCrsPair(HloInstruction* all_reduce, HloInstruction* cross_replica_sum, in ArCrsPair()
93 : ar(all_reduce), crs(cross_replica_sum), distance(dist) {} in ArCrsPair()
|
| D | hlo_reachability.cc | 124 for (HloInstruction* all_reduce : it->second) { in Build()
125 add_dependencies(all_reduce); in Build()
|
| D | bfloat16_propagation_test.cc | 211 HloInstruction* all_reduce = in TEST_F() local
216 HloInstruction::CreateGetTupleElement(shape, all_reduce, 0)); in TEST_F()
218 HloInstruction::CreateGetTupleElement(shape, all_reduce, 1)); in TEST_F()
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_utils.py | 24 from tensorflow.python.distribute import all_reduce
366 reduce_op = collective_ops.all_reduce(
393 summed_grads = all_reduce.build_ring_all_reduce(
396 summed_grads = all_reduce.build_nccl_then_ring(scaled_grads, num_shards,
399 summed_grads = all_reduce.build_nccl_then_recursive_hd(
402 summed_grads = all_reduce.build_nccl_then_shuffle(
406 summed_grads = all_reduce.build_shuffle_then_shuffle(
409 summed_grads = all_reduce.build_shuffle_all_reduce(
|
| D | BUILD | 14 name = "all_reduce",
16 "all_reduce.py",
31 ":all_reduce",
73 ":all_reduce",
|
| D | distribute_lib.py | 1475 def all_reduce(self, reduce_op, value): member in ReplicaContext
1511 return ys, lambda *dy_s: self.all_reduce(reduce_op, dy_s)
|
| D | all_reduce_test.py | 26 from tensorflow.python.distribute import all_reduce as ar
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops_test.py | 38 colred0 = collective_ops.all_reduce(in0, 2, group_key, instance_key,
42 colred1 = collective_ops.all_reduce(in1, 2, group_key, instance_key,
63 all_reduces.append(collective_ops.all_reduce(
|
| D | collective_ops.py | 24 def all_reduce(t, group_size, group_key, instance_key, merge_op, final_op, function
|
| /external/tensorflow/tensorflow/core/api_def/python_api/ |
| D | api_def_CollectiveReduce.pbtxt | 4 name: "collective.all_reduce"
|
| /external/tensorflow/tensorflow/contrib/all_reduce/python/ |
| D | all_reduce.py | 22 from tensorflow.python.distribute.all_reduce import *
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nccl_all_reduce_thunk.h | 46 const HloInstruction* all_reduce);
|
| D | nccl_all_reduce_thunk.cc | 348 const HloInstruction* all_reduce) in NcclAllReduceThunk() argument
349 : Thunk(Thunk::kNcclAllReduce, all_reduce), in NcclAllReduceThunk()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.distribute.-replica-context.pbtxt | 26 name: "all_reduce"
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.distribute.-replica-context.pbtxt | 26 name: "all_reduce"
|
| /external/tensorflow/ |
| D | CODEOWNERS | 16 # NEED OWNER: /tensorflow/contrib/all_reduce
|
| D | RELEASE.md | 562 * Add support for scalars in `tf.contrib.all_reduce`.
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | strategy_test_lib.py | 567 return ctx.all_reduce(reduce_util.ReduceOp.SUM, value)
572 return ctx.all_reduce(reduce_util.ReduceOp.MEAN, value)
|
| /external/tensorflow/tensorflow/contrib/ |
| D | BUILD | 24 "//tensorflow/contrib/all_reduce",
|
| /external/tensorflow/tensorflow/contrib/cmake/ |
| D | python_modules.txt | 84 tensorflow/contrib/all_reduce
85 tensorflow/contrib/all_reduce/python
|