p60419572
/
InfiniTensor
forked from jiuyuan/InfiniTensor
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix: 删除单独的拷贝函数 

Signed-off-by: YdrMaster <ydrml@hotmail.com>
This commit is contained in:
YdrMaster 2023-03-22 10:13:06 +08:00
parent 64a5de51f3
commit aff2b538ce
1 changed files with 0 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
pyinfinitensor/src/pyinfinitensor/onnx.py
View File

@ -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)