Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/profiler/internal/cpu/
Dhost_tracer.cc60 auto step_stats_collector = in CollectData() local
62 return CollectDataToCollector(step_stats_collector.get()); in CollectData()
66 StepStatsCollector* step_stats_collector) { in CollectDataToCollector() argument
82 step_stats_collector->SaveThreadName(cpu_name, thread.thread.tid, in CollectDataToCollector()
109 step_stats_collector->Save(cpu_name, ns); in CollectDataToCollector()
114 step_stats_collector->Finalize(); in CollectDataToCollector()
Dhost_tracer.h48 Status CollectDataToCollector(StepStatsCollector* step_stats_collector);
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.cc178 StepStatsCollector* step_stats_collector, in UpdateStats() argument
202 step_stats_collector->Finalize(); in UpdateStats()
256 std::unique_ptr<StepStatsCollector> step_stats_collector; in Run() local
258 step_stats_collector.reset(new StepStatsCollector(step_stats)); in Run()
260 params.stats_collector = step_stats_collector.get(); in Run()
299 UpdateStats(&context, step_stats_collector.get(), stats); in Run()
329 std::unique_ptr<StepStatsCollector> step_stats_collector; in Run() local
331 step_stats_collector.reset(new StepStatsCollector(step_stats)); in Run()
333 opts.stats_collector = step_stats_collector.get(); in Run()
353 if (step_stats_collector != nullptr) { in Run()
[all …]
/external/tensorflow/tensorflow/core/profiler/internal/gpu/
Dtracer.cc48 auto step_stats_collector = in CollectData() local
50 Status s = device_tracer_->Collect(step_stats_collector.get()); in CollectData()
51 step_stats_collector->Finalize(); in CollectData()
/external/tensorflow/tensorflow/core/common_runtime/
Dstep_stats_collector.h98 StepStatsCollector* step_stats_collector);
102 StepStatsCollector* step_stats_collector);
Dstep_stats_collector.cc45 const Node* node, StepStatsCollector* step_stats_collector) in NodeExecStatsWrapper() argument
47 step_stats_collector) { in NodeExecStatsWrapper()
53 StepStatsCollector* step_stats_collector) in NodeExecStatsWrapper() argument
56 step_stats_collector_(step_stats_collector) {} in NodeExecStatsWrapper()
/external/tensorflow/tensorflow/core/kernels/data/
Dcaptured_function.cc60 explicit SimpleNodeExecStats(SimpleStepStatsCollector* step_stats_collector) in SimpleNodeExecStats() argument
61 : step_stats_collector_(step_stats_collector) {} in SimpleNodeExecStats()
/external/tensorflow/tensorflow/core/
DBUILD3071 "common_runtime/step_stats_collector.h",
3136 "common_runtime/step_stats_collector.cc",