Searched refs:recv_done_instr (Results 1 – 1 of 1) sorted by relevance
3228 HloInstructionProto recv_done_instr; in RecvWithToken() local3229 *recv_done_instr.mutable_shape() = in RecvWithToken()3232 recv_done_instr.set_channel_id(handle.handle()); in RecvWithToken()3233 return AddInstruction(std::move(recv_done_instr), HloOpcode::kRecvDone, in RecvWithToken()3315 HloInstructionProto recv_done_instr; in RecvFromHost() local3316 *recv_done_instr.mutable_shape() = in RecvFromHost()3319 recv_done_instr.set_channel_id(handle.handle()); in RecvFromHost()3320 recv_done_instr.set_is_host_transfer(true); in RecvFromHost()3321 return AddInstruction(std::move(recv_done_instr), HloOpcode::kRecvDone, in RecvFromHost()