Home
last modified time | relevance | path

Searched refs:HloAllToAllInstruction (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_instructions.cc587 HloAllToAllInstruction::HloAllToAllInstruction( in HloAllToAllInstruction() function in xla::HloAllToAllInstruction
594 HloAllToAllInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()
597 return absl::make_unique<HloAllToAllInstruction>(shape, new_operands, in CloneWithNewOperandsImpl()
Dhlo_instructions.h357 class HloAllToAllInstruction : public HloCollectiveInstruction {
359 explicit HloAllToAllInstruction(
Dhlo_instruction.cc845 return absl::make_unique<HloAllToAllInstruction>(shape, operands, in CreateAllToAll()