From aff2b538cebb72b04d91c69a47849c4237cc99df Mon Sep 17 00:00:00 2001 From: YdrMaster Date: Wed, 22 Mar 2023 10:13:06 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E5=88=A0=E9=99=A4=E5=8D=95=E7=8B=AC?= =?UTF-8?q?=E7=9A=84=E6=8B=B7=E8=B4=9D=E5=87=BD=E6=95=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: YdrMaster --- pyinfinitensor/src/pyinfinitensor/onnx.py | 12 ------------ 1 file changed, 12 deletions(-) diff --git a/pyinfinitensor/src/pyinfinitensor/onnx.py b/pyinfinitensor/src/pyinfinitensor/onnx.py index e9424ab3..9a6afd21 100644 --- a/pyinfinitensor/src/pyinfinitensor/onnx.py +++ b/pyinfinitensor/src/pyinfinitensor/onnx.py @@ -578,18 +578,6 @@ class OnnxStub: def run(self) -> None: self.handler.run() - def put_int32(self, name: str, data: List[int]) -> None: - self.handler.copy_int32(self.inputs[name], data) - - def put_int64(self, name: str, data: List[int]) -> None: - self.handler.copy_int64(self.inputs[name], data) - - def put_float(self, name: str, data: List[int]) -> None: - self.handler.copy_float(self.inputs[name], data) - - def take_float(self) -> List[float]: - return next(iter(self.outputs.values())).cloneFloats() - def from_onnx(model: ModelProto, runtime): stub = OnnxStub(model, runtime)