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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -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.