Searched defs:step_stats_collector (Results 1 – 4 of 4) sorted by relevance
104 StepStatsCollector step_stats_collector(run_metadata->mutable_step_stats()); in CollectData() local
54 const NodeDef* node, StepStatsCollector* step_stats_collector) in NodeExecStatsWrapper()62 StepStatsCollector* step_stats_collector) in NodeExecStatsWrapper()
579 StepStatsCollector step_stats_collector(&step_stats_); in CollectData() local
66 explicit SimpleNodeExecStats(SimpleStepStatsCollector* step_stats_collector) in SimpleNodeExecStats()