Searched refs:send_instr (Results 1 – 1 of 1) sorted by relevance
3396 HloInstructionProto send_instr; in SendWithToken() local3398 *send_instr.mutable_shape() = in SendWithToken()3402 send_instr.set_channel_id(handle.handle()); in SendWithToken()3404 AddInstruction(std::move(send_instr), HloOpcode::kSend, in SendWithToken()3494 HloInstructionProto send_instr; in SendToHost() local3495 *send_instr.mutable_shape() = in SendToHost()3500 send_instr.set_channel_id(handle.handle()); in SendToHost()3501 send_instr.set_is_host_transfer(true); in SendToHost()3503 AddInstruction(std::move(send_instr), HloOpcode::kSend, in SendToHost()