INNER CODE UNIT · Python

handle

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

            handle = counter + 1
            models[handle] = runner
            return {"handle": handle}, []
        if op == "unload":
            _close(models.pop(int(header["handle"]), None))
            return {}, []
        if op == "run":
            runner = self._model(models, header)
            outputs = runner.run(proto.decode_tensors(header["tensors"], blobs))
            specs, out_blobs = proto.encode_tensors(outputs)
            return {"tensors": specs}, out_blobs
        if op == "time":
            runner = self._model(models, header)
            inputs = proto.decode_tensors(header["tensors"], blobs)
            number, repeat = (
                max(int(header.get("number", 1)), 1),
                max(int(header.get("repeat", 1)), 1),
            )

View source record →