Searched refs:recv_done_instr (Results 1 – 1 of 1) sorted by relevance
2396 HloInstructionProto recv_done_instr; in RecvWithToken() local2397 *recv_done_instr.mutable_shape() = in RecvWithToken()2400 recv_done_instr.set_channel_id(handle.handle()); in RecvWithToken()2401 return AddInstruction(std::move(recv_done_instr), HloOpcode::kRecvDone, in RecvWithToken()2483 HloInstructionProto recv_done_instr; in RecvFromHost() local2484 *recv_done_instr.mutable_shape() = in RecvFromHost()2487 recv_done_instr.set_channel_id(handle.handle()); in RecvFromHost()2488 recv_done_instr.set_is_host_transfer(true); in RecvFromHost()2489 return AddInstruction(std::move(recv_done_instr), HloOpcode::kRecvDone, in RecvFromHost()