From 1a22ddc4e545ec76c0b7876788059f5a69406bca Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Sat, 16 Mar 2019 12:26:07 -0700 Subject: [PATCH] Show op versions is tflite model visualization tool. PiperOrigin-RevId: 238810149 --- tensorflow/lite/tools/visualize.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tensorflow/lite/tools/visualize.py b/tensorflow/lite/tools/visualize.py index 53bb67e3fed..e0dd258ebb9 100644 --- a/tensorflow/lite/tools/visualize.py +++ b/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