Searched refs:send_instr (Results 1 – 1 of 1) sorted by relevance
3166 HloInstructionProto send_instr; in SendWithToken() local3168 *send_instr.mutable_shape() = in SendWithToken()3172 send_instr.set_channel_id(handle.handle()); in SendWithToken()3174 AddInstruction(std::move(send_instr), HloOpcode::kSend, in SendWithToken()3264 HloInstructionProto send_instr; in SendToHost() local3265 *send_instr.mutable_shape() = in SendToHost()3270 send_instr.set_channel_id(handle.handle()); in SendToHost()3271 send_instr.set_is_host_transfer(true); in SendToHost()3273 AddInstruction(std::move(send_instr), HloOpcode::kSend, in SendToHost()