Skip to content
Permalink
Browse files

Show op versions is tflite model visualization tool.

PiperOrigin-RevId: 238810149
  • Loading branch information...
tensorflower-gardener committed Mar 16, 2019
1 parent ebc91ed commit 1a22ddc4e545ec76c0b7876788059f5a69406bca
Showing with 2 additions and 1 deletion.
  1. +2 −1 tensorflow/lite/tools/visualize.py
@@ -339,7 +339,8 @@ def CreateHtmlFile(tflite_input, html_output):

# Spec on what keys to display
buffer_keys_to_display = [("data", DataSizeMapper())]
operator_keys_to_display = [("builtin_code", None), ("custom_code", None)]
operator_keys_to_display = [("builtin_code", None), ("custom_code", None),
("version", None)]

for subgraph_idx, g in enumerate(data["subgraphs"]):
# Subgraph local specs on what to display

0 comments on commit 1a22ddc

Please sign in to comment.
You can’t perform that action at this time.