Searched refs:send_done_instr (Results 1 – 1 of 1) sorted by relevance
2375 HloInstructionProto send_done_instr; in SendWithToken() local2376 *send_done_instr.mutable_shape() = ShapeUtil::MakeTokenShape().ToProto(); in SendWithToken()2377 send_done_instr.set_channel_id(handle.handle()); in SendWithToken()2378 return AddInstruction(std::move(send_done_instr), HloOpcode::kSendDone, in SendWithToken()2474 HloInstructionProto send_done_instr; in SendToHost() local2475 *send_done_instr.mutable_shape() = ShapeUtil::MakeTokenShape().ToProto(); in SendToHost()2476 send_done_instr.set_channel_id(handle.handle()); in SendToHost()2477 send_done_instr.set_is_host_transfer(true); in SendToHost()2478 return AddInstruction(std::move(send_done_instr), HloOpcode::kSendDone, in SendToHost()