We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent cb7dfaa commit aefffa8Copy full SHA for aefffa8
scripts/build_iree.py
@@ -13,7 +13,8 @@
13
from pathlib import Path
14
import os
15
from kenning.utils.resource_manager import ResourceURI
16
-from kenning.optimizers.onnx import tfliteconversion, onnx
+import onnx
17
+from kenning.converters.tflite_converter import TFLiteConverter
18
19
def main():
20
parser = argparse.ArgumentParser(__doc__)
@@ -65,8 +66,9 @@ def main():
65
66
output_names = [spec["name"] for spec in io_spec["output"]]
67
except KeyError:
68
output_names = None
- model = tfliteconversion(
69
- args.input_model_path, io_spec["input"], output_names
+ model = TFLiteConverter(args.input_model_path).to_onnx(
70
+ io_spec["input"],
71
+ output_names
72
)
73
74
# Conversion from tflite to onnx changes name of the entry function.
0 commit comments