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.cc327 std::unordered_map<string, NodeExecStats> name_to_hw_node_stats; in BuildCostModel() local
339 if (name_to_hw_node_stats.find(node_name) != in BuildCostModel()
340 name_to_hw_node_stats.end()) { in BuildCostModel()
341 int64 time = name_to_hw_node_stats[node_name].op_end_rel_micros(); in BuildCostModel()
342 name_to_hw_node_stats[node_name].set_op_end_rel_micros( in BuildCostModel()
345 name_to_hw_node_stats.emplace(node_name, node_stats); in BuildCostModel()
372 if (dev_stats.hardware_stats && name_to_hw_node_stats.find(node_name) != in BuildCostModel()
373 name_to_hw_node_stats.end()) { in BuildCostModel()
374 const NodeExecStats& hw_stats = name_to_hw_node_stats[node_name]; in BuildCostModel()