INNER CODE UNIT · Python

model_bytes_to_proto

onnxsim/onnxsim · onnxsim/rpc/server.py:319

def model_bytes_to_proto(data: bytes) -> onnx.ModelProto:
    model = onnx.ModelProto()
    model.ParseFromString(data)
    return model


class _Handler(socketserver.BaseRequestHandler):
    server: "RPCServer"

    def handle(self) -> None:
        sock = self.request
        models: Dict[int, Any] = {}
        counter = 0
        try:
            first, _ = proto.recv_message(sock, self.server.max_blob_bytes)
            client_key = first.get("key", "")
            if first.get("op") != "hello" or (
                self.server.key and client_key != self.server.key

View source record →