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

More accurate input-pipeline analysis for TPU.

Browse Source 
PiperOrigin-RevId: 286672368
Change-Id: I14e6e47e79304154404629295b6728857583d781
This commit is contained in:
A. Unique TensorFlower 2019-12-20 18:58:10 -08:00 committed by TensorFlower Gardener
parent a25c899dcc
commit 801b09624f
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tensorflow/core/profiler/utils/event_span.cc
View File

@ -116,17 +116,17 @@ EventType ClassifyGpuEvent(absl::string_view event_name) {
}
EventType ClassifyCpuEvent(absl::string_view event_name, int64 correlation_id) {
if (absl::StartsWithIgnoreCase(event_name, "MEMCPYHtoD"))
if (absl::StartsWithIgnoreCase(event_name, "MEMCPYHtoD") ||
absl::StrContains(event_name, "Infeed"))
return HOST_TO_DEVICE;
if (absl::StartsWithIgnoreCase(event_name, "MEMCPYHtoH")) return HOST_TO_HOST;
if (correlation_id >= 0 ||
absl::StartsWithIgnoreCase(event_name, "ExecutorState::Process")) {
return HOST_PREPARE;
} else {
if (absl::StartsWithIgnoreCase(event_name, "IteratorGetNext"))
return HOST_WAIT_INPUT;
return HOST_COMPUTE;
}
if (absl::StartsWithIgnoreCase(event_name, "IteratorGetNext"))
return HOST_WAIT_INPUT;
return HOST_COMPUTE;
}
std::string PrintEventType(EventType event_type) {