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

Add optimization profile type in the HLO metadata

Browse Source 
PiperOrigin-RevId: 336180647
Change-Id: I53e7a0ff14d61c13af25dbbf53996d2f556300a4
This commit is contained in:
A. Unique TensorFlower 2020-10-08 15:47:52 -07:00 committed by TensorFlower Gardener
parent 101afeb9b9
commit d1369b5c09
1 changed files with 9 additions and 0 deletions

9
tensorflow/compiler/xla/xla_data.proto
View File

@ -245,6 +245,13 @@ message ComputationStats {
double transcendental_count = 2; double transcendental_count = 2;
} }
// The type optimization profiles in use.
enum ProfileType {
INVALID = 0;
WINDOW = 1;
FLAG = 2;
}
// Symbolization metadata for HLO Instructions. // Symbolization metadata for HLO Instructions.
// //
// This metadata is used for debugging XLA code generation, as well as // 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. // e.g. it could be the file and line of user code that generated the op.
string source_file = 3; string source_file = 3;
int32 source_line = 4; int32 source_line = 4;
repeated ProfileType profile_type = 5;
} }
// Profile data from the execution of a computation. // Profile data from the execution of a computation.