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

Add more StatType.

Browse Source 
PiperOrigin-RevId: 288720224
Change-Id: Ieeed4dfcacd5ed435068aa17808d29f3300470bb
This commit is contained in:
A. Unique TensorFlower 2020-01-08 10:05:44 -08:00 committed by TensorFlower Gardener
parent 2b3296441b
commit 2ad9dd652f
2 changed files with 13 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tensorflow/core/profiler/utils/xplane_schema.cc
View File

@ -68,11 +68,13 @@ static const absl::string_view kStatTypeStrMap[] = {
"device_ordinal", "chip_ordinal",
"node_ordinal", "model_id",
"queue_addr", "request_id",
"run_id", "correlation_id",
"graph_type", "step_num",
"iter_num", "index_on_host",
"bytes_reserved", "bytes_allocated",
"bytes_available", "fragmentation",
"run_id", "graph_type",
"step_num", "iter_num",
"index_on_host", "bytes_reserved",
"bytes_allocated", "bytes_available",
"fragmentation", "device_id",
"context_id", "correlation_id",
"memcpy_details", "memalloc_details",
"kernel_details", "group_id",
"step_name", "level 0",
"tf_op", "hlo_op",

7
tensorflow/core/profiler/utils/xplane_schema.h
View File

@ -74,7 +74,6 @@ enum StatType {
kQueueAddr,
kRequestId,
kRunId,
kCorrelationId,
kGraphType,
kStepNum,
kIterNum,
@ -83,6 +82,12 @@ enum StatType {
kBytesAllocated,
kBytesAvailable,
kFragmentation,
// Device trace arguments.
kDeviceId,
kContextId,
kCorrelationId,
kMemcpyDetails,
kMemallocDetails,
kKernelDetails,
// Stats added when processing traces.
kGroupId,