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)