Home
last modified time | relevance | path

Searched refs:op_info_map_ (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/profiler/device/cpu/
Dcpu_profiling.cc48 auto iter = op_info_map_.find(op_name); in SetRunTimeData()
49 if (iter != op_info_map_.end()) { in SetRunTimeData()
56 op_info_map_[op_name] = op_info; in SetRunTimeData()
100 cpu_data_saver_inst->ParseOpInfo(op_info_map_); in SaveProfileData()
105 void CPUProfiler::ClearInst() { op_info_map_.clear(); } in ClearInst()
/third_party/mindspore/mindspore/ccsrc/profiler/device/
Dprofiling.cc55 auto iter = op_info_map_.find(op_name); in SetRunTimeData()
56 if (iter != op_info_map_.end()) { in SetRunTimeData()
62 auto iter = op_info_map_.find(op_name); in SetRunTimeData()
63 if (iter != op_info_map_.end()) { in SetRunTimeData()
Dprofiling.h97 std::unordered_map<std::string, OpInfo> op_info_map_; variable
/third_party/mindspore/mindspore/ccsrc/profiler/device/gpu/
Dgpu_profiling.cc296 auto iter = op_info_map_.find(event.op_name); in ProcessEvents()
297 if (iter != op_info_map_.end()) { in ProcessEvents()
339 std::vector<std::pair<std::string, OpInfo>> order_vec(op_info_map_.begin(), op_info_map_.end()); in OpsParser()
421 auto iter = op_info_map_.find(op_name); in SetRunTimeData()
422 if (iter != op_info_map_.end()) { in SetRunTimeData()
429 op_info_map_[op_name] = op_info; in SetRunTimeData()
516 dataSaver.ParseOpInfo(op_info_map_); in SaveProfileData()
524 op_info_map_.clear(); in ClearInst()