Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_node_show.cc60 GraphNodeProto* node_pb = node->mutable_proto(); in AggregateTotalStats() local
62 node_pb->total_run_count()); in AggregateTotalStats()
64 proto().total_definition_count() + node_pb->total_definition_count()); in AggregateTotalStats()
66 node_pb->total_exec_micros()); in AggregateTotalStats()
69 node_pb->total_accelerator_exec_micros()); in AggregateTotalStats()
71 node_pb->total_cpu_exec_micros()); in AggregateTotalStats()
74 node_pb->total_requested_bytes()); in AggregateTotalStats()
76 node_pb->total_peak_bytes()); in AggregateTotalStats()
78 node_pb->total_residual_bytes()); in AggregateTotalStats()
80 node_pb->total_output_bytes()); in AggregateTotalStats()
[all …]
Dtfprof_stats.cc95 for (const auto& node_pb : profile.nodes()) { in TFStats() local
97 new TFGraphNode(node_pb.second, profile, &id_to_string_, &nodes_map_)); in TFStats()
99 node_pb.second.name(), std::move(node))); in TFStats()