Skip to content

Commit 18a396a

Browse files
committed
Add target kwargs to api
1 parent 357a212 commit 18a396a

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

utensor_cgen/api/export.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,8 @@ def tflm_keras_export(
1111
representive_dataset,
1212
model_name=None,
1313
optimizations=None,
14-
config_file='utensor_cli.toml'
14+
config_file='utensor_cli.toml',
15+
target='utensor',
1516
):
1617
with tempfile.TemporaryDirectory(prefix='utensor_') as tmp_dir:
1718
dir_path = Path(tmp_dir)
@@ -36,4 +37,4 @@ def tflm_keras_export(
3637
with (dir_path / 'tflm_model.tflite').open('wb') as fid:
3738
fid.write(tflm_model_content)
3839
fid.flush()
39-
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

Comments
 (0)