Searched refs:recv_done_instr (Results 1 – 1 of 1) sorted by relevance
3458 HloInstructionProto recv_done_instr; in RecvWithToken() local3459 *recv_done_instr.mutable_shape() = in RecvWithToken()3462 recv_done_instr.set_channel_id(handle.handle()); in RecvWithToken()3463 return AddInstruction(std::move(recv_done_instr), HloOpcode::kRecvDone, in RecvWithToken()3547 HloInstructionProto recv_done_instr; in RecvFromHost() local3548 *recv_done_instr.mutable_shape() = in RecvFromHost()3551 recv_done_instr.set_channel_id(handle.handle()); in RecvFromHost()3552 recv_done_instr.set_is_host_transfer(true); in RecvFromHost()3553 return AddInstruction(std::move(recv_done_instr), HloOpcode::kRecvDone, in RecvFromHost()