Home
last modified time | relevance | path

Searched defs:all_to_all (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dall_to_all_decomposer.cc39 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
Dhlo_matchers_test.cc337 std::unique_ptr<HloInstruction> all_to_all = in TEST_F() local
Dhlo_verifier.cc487 auto* all_to_all = Cast<HloAllToAllInstruction>(hlo); in HandleAllToAll() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_layout_assignment.cc391 auto* all_to_all = Cast<HloAllToAllInstruction>(instruction); in AddBackendConstraints() local
/external/tensorflow/tensorflow/python/tpu/ops/
Dtpu_ops.py40 def all_to_all(x, function
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dfft_handler.cc165 auto all_to_all = collective_ops_creator.create_cross_partition_all_to_all( in ShuffleDataWithAllToAll() local
Dspmd_partitioner_test.cc1178 auto all_to_all = in TEST_F() local
1624 auto all_to_all = in TEST_F() local
1659 auto all_to_all = in TEST_F() local
7319 auto all_to_all = AllOf(op::Shape("f32[4,4,2,4,4]"), op::AllToAll(reshape)); in TEST_F() local
7343 auto all_to_all = op::AllToAll( in TEST_F() local
7366 auto all_to_all = op::AllToAll( in TEST_F() local
7636 auto all_to_all = AllOf(op::Shape("f32[2,2,4,12,32]"), op::AllToAll(reshape)); in TEST_F() local
Dspmd_partitioner.cc1585 auto all_to_all = in ReshardWithAllToAll() local
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dhlo_function_importer.cc1008 auto all_to_all = Cast<HloAllToAllInstruction>(instruction); in ImportInstructionImpl() local
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc1147 auto* all_to_all = xla::Cast<xla::HloAllToAllInstruction>(instr); in EmitAllToAllOp() local