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

Remove the old grouping rule for PJRT.

Browse Source 
PiperOrigin-RevId: 317170383
Change-Id: I96973a2d2cd0ca1fc786bc7500deb6b4fedd0534
This commit is contained in:
Jiho Choi 2020-06-18 13:35:31 -07:00 committed by TensorFlower Gardener
parent 6558da5a66
commit e0962f4c37
3 changed files with 1 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/core/profiler/utils/group_events.cc
View File

@ -635,10 +635,7 @@ std::vector<InterThreadConnectInfo> CreateInterThreadConnectInfoList() {
{StatType::kStepId, StatType::kIterNum}}, {StatType::kStepId, StatType::kIterNum}},
{HostEventType::kKernelLaunch, {HostEventType::kKernelLaunch,
HostEventType::kKernelExecute, HostEventType::kKernelExecute,
{StatType::kCorrelationId}}, {StatType::kCorrelationId}}};
{HostEventType::kLocalExecutableExecuteOnLocalDevice,
HostEventType::kLocalExecutableExecute,
{StatType::kRunId}}};
return connect_info_list; return connect_info_list;
} }

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

@ -91,10 +91,6 @@ const HostEventTypeMap& GetHostEventTypeMap() {
{"WhileOp-StartBody", kWhileOpStartBody}, {"WhileOp-StartBody", kWhileOpStartBody},
{"ForOp", kForOp}, {"ForOp", kForOp},
{"PartitionedCallOp", kPartitionedCallOp}, {"PartitionedCallOp", kPartitionedCallOp},
// XLA related.
{"LocalExecutable::ExecuteOnLocalDevices",
kLocalExecutableExecuteOnLocalDevice},
{"LocalExecutable::Execute", kLocalExecutableExecute},
// tf.data related. // tf.data related.
{"IteratorGetNextOp::DoCompute", kIteratorGetNextOp}, {"IteratorGetNextOp::DoCompute", kIteratorGetNextOp},
{"IteratorGetNextAsOptionalOp::DoCompute", kIteratorGetNextAsOptionalOp}, {"IteratorGetNextAsOptionalOp::DoCompute", kIteratorGetNextAsOptionalOp},

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

@ -81,9 +81,6 @@ enum HostEventType {
kWhileOpStartBody, kWhileOpStartBody,
kForOp, kForOp,
kPartitionedCallOp, kPartitionedCallOp,
// XLA related.
kLocalExecutableExecuteOnLocalDevice,
kLocalExecutableExecute,
// tf.data related. // tf.data related.
kIteratorGetNextOp, kIteratorGetNextOp,
kIteratorGetNextAsOptionalOp, kIteratorGetNextAsOptionalOp,