Searched defs:all_to_all (Results 1 – 7 of 7) 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
86 HloInstruction* all_to_all = in ExpandInstruction() local
|
| D | hlo_verifier.cc | 292 auto* all_to_all = Cast<HloAllToAllInstruction>(hlo); in HandleAllToAll() local
|
| /external/tensorflow/tensorflow/python/tpu/ops/ |
| D | tpu_ops.py | 45 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
1586 auto all_to_all = in TEST_F() local
1621 auto all_to_all = in TEST_F() local
4939 auto all_to_all = AllOf(op::Shape("f32[4,4,2,4,4]"), op::AllToAll(reshape)); in TEST_F() local
4963 auto all_to_all = op::AllToAll( in TEST_F() local
4986 auto all_to_all = op::AllToAll( in TEST_F() local
5257 auto all_to_all = AllOf(op::Shape("f32[2,2,4,12,32]"), op::AllToAll(reshape)); in TEST_F() local
|
| D | spmd_partitioner.cc | 1184 auto all_to_all = in ReshardWithAllToAll() local
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 1051 auto* all_to_all = xla::Cast<xla::HloAllToAllInstruction>(instr); in EmitAllToAllOp() local
|