| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | all_reduce_simplifier.cc | 39 [this](const HloInstruction* all_reduce) -> int64 { in Run()
80 auto all_reduce = all_reduce_and_group_size.first; in Run() local
|
| D | while_loop_all_reduce_code_motion.cc | 66 MovableAllReduceContext IsAllReduceMovable(HloInstruction* all_reduce, in IsAllReduceMovable()
68 auto all_reduce_is_summation = [](HloInstruction* all_reduce) -> bool { in IsAllReduceMovable()
272 HloComputation* while_body) -> MovableAllReduceContext { in IsAllReduceMovable()
592 HloInstruction* all_reduce = all_reduce_accumulations_pair.first; in Run() local
|
| D | ar_crs_combiner.cc | 96 auto all_reduce = Cast<HloAllReduceInstruction>(hlo); in HasCombinableReplicaGroup() local
543 auto all_reduce = pair.ar; in RewriteGraph() local
|
| D | all_reduce_combiner_test.cc | 221 auto all_reduce = b.AddInstruction(HloInstruction::CreateAllReduce( in TEST_F() local
|
| D | hlo_cost_analysis_test.cc | 852 HloInstruction* all_reduce = module->entry_computation()->root_instruction(); in TEST_F() local
|
| D | bfloat16_propagation_test.cc | 211 HloInstruction* all_reduce = in TEST_F() local
|
| D | hlo_verifier.cc | 1869 auto all_reduce = DynCast<HloAllReduceInstruction>(instruction); in VerifyLayoutConstrainedAllReduce() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nvptx_compiler_test.cc | 62 HloInstruction* all_reduce = module->entry_computation()->root_instruction(); in TEST_F() local
103 HloInstruction* all_reduce = module->entry_computation()->root_instruction(); in TEST_F() local
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | collective_ops_test.py | 47 all_reduce = _collective_ops.all_reduce variable in CollectiveOpsV1
56 def all_reduce(t, group_size, group_key, instance_key, *args, **kwargs): member in CollectiveOpsV2
471 def all_reduce(device): function
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops.py | 23 def all_reduce(t, function
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_utils.py | 335 def all_reduce(self, member in CollectiveReplicaLauncher
|
| D | vars_test.py | 999 def all_reduce(): function
|
| D | tpu_strategy_test.py | 692 def all_reduce(x): function
|
| D | distribute_lib.py | 3167 def all_reduce(self, reduce_op, value, options=None): member in ReplicaContextBase
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 1119 auto* all_reduce = xla::Cast<xla::HloAllReduceInstruction>(instr); in EmitAllReduceOp() local
1145 auto* all_reduce = xla::Cast<xla::HloAllReduceInstruction>(instr); in EmitAllReduceStartOp() local
|
| D | legalize_tf.cc | 6393 auto all_reduce = rewriter.create<AllReduceOp>( in matchAndRewrite() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | gather_scatter_handler.cc | 624 auto all_reduce = operand.state().partitioner->AllReduceAlongShardingDims( in HandleScatter() local
|
| D | spmd_partitioner.cc | 1674 auto all_reduce = per_group_partitioner_state.collective_ops_creator in HandleConcatenate() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 577 auto all_reduce = Cast<HloAllReduceInstruction>(instruction); in ImportInstructionImpl() local
|