Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dstep_stats_collector.cc325 std::unordered_map<string, NodeExecStats> name_to_hw_node_stats; in BuildCostModel() local
337 if (name_to_hw_node_stats.find(node_name) != in BuildCostModel()
338 name_to_hw_node_stats.end()) { in BuildCostModel()
339 int64 time = name_to_hw_node_stats[node_name].op_end_rel_micros(); in BuildCostModel()
340 name_to_hw_node_stats[node_name].set_op_end_rel_micros( in BuildCostModel()
343 name_to_hw_node_stats.emplace(node_name, node_stats); in BuildCostModel()
370 if (dev_stats.hardware_stats && name_to_hw_node_stats.find(node_name) != in BuildCostModel()
371 name_to_hw_node_stats.end()) { in BuildCostModel()
372 const NodeExecStats& hw_stats = name_to_hw_node_stats[node_name]; in BuildCostModel()