saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-24 02:47:54 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: import error in TrtTable::Dict method (#8940)

Browse source
This commit is contained in:
senysenyseny16 2021-09-07 14:28:49 +07:00 committed by GitHub
parent 5d47b2e431
commit 3be96f8a15
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/python/tools/quantization/CalTableFlatBuffers/TrtTable.py
View file

@ -28,7 +28,7 @@ class TrtTable(object):
x = self._tab.Vector(o)
x += flatbuffers.number_types.UOffsetTFlags.py_type(j) * 4
x = self._tab.Indirect(x)
from CalTableFlatBuffers.KeyValue import KeyValue
from onnxruntime.quantization.CalTableFlatBuffers.KeyValue import KeyValue
obj = KeyValue()
obj.Init(self._tab.Bytes, x)
return obj