Searched refs:send_done_instr (Results 1 – 1 of 1) sorted by relevance
2345 HloInstructionProto send_done_instr; in SendWithToken() local2346 *send_done_instr.mutable_shape() = ShapeUtil::MakeTokenShape().ToProto(); in SendWithToken()2347 send_done_instr.set_channel_id(handle.handle()); in SendWithToken()2348 return AddInstruction(std::move(send_done_instr), HloOpcode::kSendDone, in SendWithToken()2444 HloInstructionProto send_done_instr; in SendToHost() local2445 *send_done_instr.mutable_shape() = ShapeUtil::MakeTokenShape().ToProto(); in SendToHost()2446 send_done_instr.set_channel_id(handle.handle()); in SendToHost()2447 send_done_instr.set_is_host_transfer(true); in SendToHost()2448 return AddInstruction(std::move(send_done_instr), HloOpcode::kSendDone, in SendToHost()