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 357a212 commit 18a396aCopy full SHA for 18a396a
utensor_cgen/api/export.py
@@ -11,7 +11,8 @@ def tflm_keras_export(
11
representive_dataset,
12
model_name=None,
13
optimizations=None,
14
- config_file='utensor_cli.toml'
+ config_file='utensor_cli.toml',
15
+ target='utensor',
16
):
17
with tempfile.TemporaryDirectory(prefix='utensor_') as tmp_dir:
18
dir_path = Path(tmp_dir)
@@ -36,4 +37,4 @@ def tflm_keras_export(
36
37
with (dir_path / 'tflm_model.tflite').open('wb') as fid:
38
fid.write(tflm_model_content)
39
fid.flush()
- convert_graph(fid.name, config=config_file, model_name=model_name)
40
+ convert_graph(fid.name, config=config_file, model_name=model_name, target=target)
0 commit comments