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

fix

This commit is contained in:
wanghailu 2023-02-27 10:22:03 +08:00 committed by YdrMaster
parent 5aeacedab3
commit 64a5de51f3
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -588,7 +588,7 @@ class OnnxStub:
self.handler.copy_float(self.inputs[name], data)
def take_float(self) -> List[float]:
return next(iter(self.handler.outputs.values())).copyFloats()
return next(iter(self.outputs.values())).cloneFloats()
def from_onnx(model: ModelProto, runtime):