| 429 | } |
| 430 | |
| 431 | void StepStatsCollector::SaveThreadName(const string& device, |
| 432 | const uint32 thread_id, |
| 433 | const string& thread_name) { |
| 434 | VLOG(1) << "Save dev " << device << " thread id " << thread_id << " name " |
| 435 | << thread_name; |
| 436 | { |
| 437 | mutex_lock l(mu_); |
| 438 | if (finalized_) { |
| 439 | LOG(WARNING) << "thread_name saved after finalize will not be collected."; |
| 440 | } |
| 441 | auto& thread_names_map = thread_names_[device]; |
| 442 | thread_names_map[thread_id] = thread_name; |
| 443 | } |
| 444 | } |
| 445 | |
| 446 | NodeExecStatsInterface* StepStatsCollector::CreateNodeExecStats( |
| 447 | const NodeDef* node) { |