Searched refs:HloSendInstruction (Results 1 – 4 of 4) sorted by relevance
106 !DynCast<HloSendInstruction>(instruction)->is_host_transfer()) { in GlobalPredecessors()186 !DynCast<HloSendInstruction>(instruction)->is_host_transfer()) { in GlobalSuccessors()
397 HloSendInstruction::HloSendInstruction(HloInstruction* operand, in HloSendInstruction() function in xla::HloSendInstruction410 std::unique_ptr<HloInstruction> HloSendInstruction::CloneWithNewOperandsImpl( in CloneWithNewOperandsImpl()414 return absl::make_unique<HloSendInstruction>( in CloneWithNewOperandsImpl()418 HloSendDoneInstruction::HloSendDoneInstruction(HloSendInstruction* operand, in HloSendDoneInstruction()432 Cast<HloSendInstruction>(new_operands[0]), is_host_transfer()); in CloneWithNewOperandsImpl()
240 class HloSendInstruction : public HloSendRecvInstruction {242 explicit HloSendInstruction(HloInstruction* operand, HloInstruction* token,254 explicit HloSendDoneInstruction(HloSendInstruction* operand,
879 return absl::make_unique<HloSendInstruction>(operand, token, channel_id, in CreateSend()885 auto send_operand = DynCast<HloSendInstruction>(operand); in CreateSendDone()