From d1369b5c09e2e3e7b78c7daaa520e1ebd106eb13 Mon Sep 17 00:00:00 2001
From: "A. Unique TensorFlower" <gardener@tensorflow.org>
Date: Thu, 8 Oct 2020 15:47:52 -0700
Subject: [PATCH] Add optimization profile type in the HLO metadata
PiperOrigin-RevId: 336180647
Change-Id: I53e7a0ff14d61c13af25dbbf53996d2f556300a4
---
tensorflow/compiler/xla/xla_data.proto | 9 +++++++++
1 file changed, 9 insertions(+)
diff --git a/tensorflow/compiler/xla/xla_data.proto b/tensorflow/compiler/xla/xla_data.proto
index 2d311dd2f70..7da8d2cb84d 100644
--- a/tensorflow/compiler/xla/xla_data.proto
+++ b/tensorflow/compiler/xla/xla_data.proto
@@ -245,6 +245,13 @@ message ComputationStats {
double transcendental_count = 2;
}
+// The type optimization profiles in use.
+enum ProfileType {
+ INVALID = 0;
+ WINDOW = 1;
+ FLAG = 2;
+}
+
// Symbolization metadata for HLO Instructions.
//
// This metadata is used for debugging XLA code generation, as well as
@@ -268,6 +275,8 @@ message OpMetadata {
// e.g. it could be the file and line of user code that generated the op.
string source_file = 3;
int32 source_line = 4;
+
+ repeated ProfileType profile_type = 5;
}
// Profile data from the execution of a computation.