From 2a1ab4de133859ad968b610e90aa97a64469dd3f Mon Sep 17 00:00:00 2001
From: Lutz Roeder <lutzroeder@users.noreply.github.com>
Date: Thu, 20 Jul 2023 20:34:46 -0700
Subject: [PATCH] Update tf-proto.js

---
 source/tf-proto.js | 28 ++++++++++++++++++++++++++++
 1 file changed, 28 insertions(+)

diff --git a/source/tf-proto.js b/source/tf-proto.js
index 7694c6e8d0..37dfcc85f2 100644
--- a/source/tf-proto.js
+++ b/source/tf-proto.js
@@ -2549,7 +2549,10 @@ $root.tensorflow.GraphDebugInfo = class GraphDebugInfo {
 
     constructor() {
         this.files = [];
+        this.frames_by_id = {};
+        this.traces_by_id = {};
         this.traces = {};
+        this.name_to_trace_id = {};
     }
 
     static decode(reader, length) {
@@ -2561,9 +2564,18 @@ $root.tensorflow.GraphDebugInfo = class GraphDebugInfo {
                 case 1:
                     message.files.push(reader.string());
                     break;
+                case 4:
+                    reader.entry(message.frames_by_id, () => reader.fixed64(), () => $root.tensorflow.GraphDebugInfo.FileLineCol.decode(reader, reader.uint32()));
+                    break;
+                case 6:
+                    reader.entry(message.traces_by_id, () => reader.fixed64(), () => $root.tensorflow.GraphDebugInfo.StackTrace.decode(reader, reader.uint32()));
+                    break;
                 case 2:
                     reader.entry(message.traces, () => reader.string(), () => $root.tensorflow.GraphDebugInfo.StackTrace.decode(reader, reader.uint32()));
                     break;
+                case 5:
+                    reader.entry(message.name_to_trace_id, () => reader.string(), () => reader.fixed64());
+                    break;
                 default:
                     reader.skipType(tag & 7);
                     break;
@@ -2581,9 +2593,18 @@ $root.tensorflow.GraphDebugInfo = class GraphDebugInfo {
                 case "files":
                     reader.array(message.files, () => reader.string());
                     break;
+                case "frames_by_id":
+                    reader.entry(message.frames_by_id, () => reader.fixed64(), () => $root.tensorflow.GraphDebugInfo.FileLineCol.decodeText(reader));
+                    break;
+                case "traces_by_id":
+                    reader.entry(message.traces_by_id, () => reader.fixed64(), () => $root.tensorflow.GraphDebugInfo.StackTrace.decodeText(reader));
+                    break;
                 case "traces":
                     reader.entry(message.traces, () => reader.string(), () => $root.tensorflow.GraphDebugInfo.StackTrace.decodeText(reader));
                     break;
+                case "name_to_trace_id":
+                    reader.entry(message.name_to_trace_id, () => reader.string(), () => reader.fixed64());
+                    break;
                 default:
                     reader.field(tag, message);
                     break;
@@ -2667,6 +2688,7 @@ $root.tensorflow.GraphDebugInfo.StackTrace = class StackTrace {
 
     constructor() {
         this.file_line_cols = [];
+        this.frame_id = [];
     }
 
     static decode(reader, length) {
@@ -2678,6 +2700,9 @@ $root.tensorflow.GraphDebugInfo.StackTrace = class StackTrace {
                 case 1:
                     message.file_line_cols.push($root.tensorflow.GraphDebugInfo.FileLineCol.decode(reader, reader.uint32()));
                     break;
+                case 2:
+                    message.frame_id = reader.array(message.frame_id, () => reader.fixed64(), tag);
+                    break;
                 default:
                     reader.skipType(tag & 7);
                     break;
@@ -2695,6 +2720,9 @@ $root.tensorflow.GraphDebugInfo.StackTrace = class StackTrace {
                 case "file_line_cols":
                     message.file_line_cols.push($root.tensorflow.GraphDebugInfo.FileLineCol.decodeText(reader));
                     break;
+                case "frame_id":
+                    reader.array(message.frame_id, () => reader.fixed64());
+                    break;
                 default:
                     reader.field(tag, message);
                     break;