experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Show op versions is tflite model visualization tool.

Browse Source 
PiperOrigin-RevId: 238810149
This commit is contained in:
A. Unique TensorFlower 2019-03-16 12:26:07 -07:00 committed by TensorFlower Gardener
parent ebc91eda27
commit 1a22ddc4e5
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
tensorflow/lite/tools/visualize.py
View File

@ -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