From 1bfed17f4070b83dcd73d1f1af912e8fa520a9e0 Mon Sep 17 00:00:00 2001
From: Jaesung Chung <jaesung@google.com>
Date: Mon, 12 Oct 2020 15:22:10 -0700
Subject: [PATCH] Update visualize tool to adopt the change of builtin code
PiperOrigin-RevId: 336755842
Change-Id: I43f1ecd87baee6b8587b039a7667152a042c49ac
---
tensorflow/lite/tools/visualize.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/tensorflow/lite/tools/visualize.py b/tensorflow/lite/tools/visualize.py
index 3d22d1bb05b..a9d337bee16 100644
--- a/tensorflow/lite/tools/visualize.py
+++ b/tensorflow/lite/tools/visualize.py
@@ -452,6 +452,10 @@ def CreateHtmlFile(tflite_input, html_output):
("custom_code", None),
("version", None)]
+ # Update builtin code fields.
+ for idx, d in enumerate(data["operator_codes"]):
+ d["builtin_code"] = max(d["builtin_code"], d["deprecated_builtin_code"])
+
for subgraph_idx, g in enumerate(data["subgraphs"]):
# Subgraph local specs on what to display
html += "<div class='subgraph'>"