Searched refs:recv_instr (Results 1 – 1 of 1) sorted by relevance
3219 HloInstructionProto recv_instr; in RecvWithToken() local3220 *recv_instr.mutable_shape() = in RecvWithToken()3224 recv_instr.set_channel_id(handle.handle()); in RecvWithToken()3225 TF_ASSIGN_OR_RETURN(XlaOp recv, AddInstruction(std::move(recv_instr), in RecvWithToken()3305 HloInstructionProto recv_instr; in RecvFromHost() local3306 *recv_instr.mutable_shape() = in RecvFromHost()3310 recv_instr.set_channel_id(handle.handle()); in RecvFromHost()3311 recv_instr.set_is_host_transfer(true); in RecvFromHost()3312 TF_ASSIGN_OR_RETURN(XlaOp recv, AddInstruction(std::move(recv_instr), in RecvFromHost()