Skip to content

Commit f0b7086

Browse files
Allow tflite to cut graph if inputs/outputs are provided (#1381)
Signed-off-by: Tom Wildenhain <[email protected]>
1 parent ade6a0a commit f0b7086

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tf2onnx/convert.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -217,6 +217,9 @@ def main():
217217
args.keras, args.inputs, args.outputs)
218218
model_path = args.keras
219219
if args.tflite:
220+
# Optional, but used to cut graph if provided.
221+
inputs = args.inputs
222+
outputs = args.outputs
220223
tflite_path = args.tflite
221224
model_path = tflite_path
222225

0 commit comments

Comments
 (0)