Searched defs:all_to_all (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | all_to_all_decomposer.cc | 39 auto* all_to_all = DynCast<HloAllToAllInstruction>(instruction); in InstructionMatchesPattern() local
57 auto* all_to_all = Cast<HloAllToAllInstruction>(instruction); in ExpandInstruction() local
87 HloInstruction* all_to_all = in ExpandInstruction() local
|
| D | hlo_verifier.cc | 458 auto* all_to_all = Cast<HloAllToAllInstruction>(hlo); in HandleAllToAll() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment.cc | 339 auto* all_to_all = Cast<HloAllToAllInstruction>(instruction); in AddBackendConstraints() local
|
| /external/tensorflow/tensorflow/python/tpu/ops/ |
| D | tpu_ops.py | 44 def all_to_all(x, function
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | fft_handler.cc | 167 auto all_to_all = collective_ops_creator.create_cross_partition_all_to_all( in ShuffleDataWithAllToAll() local
|
| D | spmd_partitioner_test.cc | 1144 auto all_to_all = in TEST_F() local
1587 auto all_to_all = in TEST_F() local
1622 auto all_to_all = in TEST_F() local
5315 auto all_to_all = AllOf(op::Shape("f32[4,4,2,4,4]"), op::AllToAll(reshape)); in TEST_F() local
5339 auto all_to_all = op::AllToAll( in TEST_F() local
5362 auto all_to_all = op::AllToAll( in TEST_F() local
5633 auto all_to_all = AllOf(op::Shape("f32[2,2,4,12,32]"), op::AllToAll(reshape)); in TEST_F() local
|
| D | spmd_partitioner.cc | 1234 auto all_to_all = in ReshardWithAllToAll() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 1091 auto* all_to_all = xla::Cast<xla::HloAllToAllInstruction>(instr); in EmitAllToAllOp() local
|