Skip to content

Commit aefffa8

Browse files
committed
[#86168] scripts: adjust build_iree.py script to recent changes in converters
Signed-off-by: Bartosz Matysiak <[email protected]>
1 parent cb7dfaa commit aefffa8

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

scripts/build_iree.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,8 @@
1313
from pathlib import Path
1414
import os
1515
from kenning.utils.resource_manager import ResourceURI
16-
from kenning.optimizers.onnx import tfliteconversion, onnx
16+
import onnx
17+
from kenning.converters.tflite_converter import TFLiteConverter
1718

1819
def main():
1920
parser = argparse.ArgumentParser(__doc__)
@@ -65,8 +66,9 @@ def main():
6566
output_names = [spec["name"] for spec in io_spec["output"]]
6667
except KeyError:
6768
output_names = None
68-
model = tfliteconversion(
69-
args.input_model_path, io_spec["input"], output_names
69+
model = TFLiteConverter(args.input_model_path).to_onnx(
70+
io_spec["input"],
71+
output_names
7072
)
7173

7274
# Conversion from tflite to onnx changes name of the entry function.

0 commit comments

Comments
 (0)