Searched refs:send_instr (Results 1 – 1 of 1) sorted by relevance
2334 HloInstructionProto send_instr; in SendWithToken() local2336 *send_instr.mutable_shape() = in SendWithToken()2340 send_instr.set_channel_id(handle.handle()); in SendWithToken()2342 AddInstruction(std::move(send_instr), HloOpcode::kSend, in SendWithToken()2432 HloInstructionProto send_instr; in SendToHost() local2433 *send_instr.mutable_shape() = in SendToHost()2438 send_instr.set_channel_id(handle.handle()); in SendToHost()2439 send_instr.set_is_host_transfer(true); in SendToHost()2441 AddInstruction(std::move(send_instr), HloOpcode::kSend, in SendToHost()