| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | step_stats_collector.h | 39 class StepStatsCollector; variable
98 StepStatsCollector* step_stats_collector);
102 StepStatsCollector* step_stats_collector);
119 friend class StepStatsCollector;
135 StepStatsCollector* const step_stats_collector_; // Not owned.
161 class StepStatsCollector : public StepStatsCollectorInterface {
164 explicit StepStatsCollector(StepStats* step_stats);
|
| D | step_stats_collector.cc | 45 const Node* node, StepStatsCollector* step_stats_collector) in NodeExecStatsWrapper()
53 StepStatsCollector* step_stats_collector) in NodeExecStatsWrapper()
198 StepStatsCollector::StepStatsCollector(StepStats* step_stats) in StepStatsCollector() function in tensorflow::StepStatsCollector
261 void StepStatsCollector::BuildCostModel( in BuildCostModel()
386 void StepStatsCollector::Save(const string& device, in Save()
393 void StepStatsCollector::Save(const string& device, in Save()
413 void StepStatsCollector::SaveThreadName(const string& device, in SaveThreadName()
428 NodeExecStatsInterface* StepStatsCollector::CreateNodeExecStats( in CreateNodeExecStats()
437 string StepStatsCollector::ReportAllocsOnResourceExhausted(const string& err) { in ReportAllocsOnResourceExhausted()
516 void StepStatsCollector::Finalize() { in Finalize()
[all …]
|
| D | executor.h | 32 class StepStatsCollector; variable
|
| D | direct_session.h | 185 std::unique_ptr<StepStatsCollector> collector;
|
| D | executor_test.cc | 88 StepStatsCollector step_stats_collector_;
|
| D | direct_session.cc | 558 new StepStatsCollector(run_metadata->mutable_step_stats())); in RunInternal()
901 run_state->collector.reset(new StepStatsCollector(nullptr)); in PRunSetup()
|
| D | function_test.cc | 1020 StepStatsCollector stats_collector(&stats); in TEST_F()
1064 StepStatsCollector stats_collector(&stats); in TEST_F()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | graph_mgr.h | 43 class StepStatsCollector; variable
91 StepStatsCollector* collector,
168 StepStatsCollector* collector,
177 void BuildCostModel(Item* item, StepStatsCollector* collector,
|
| D | worker_cache_logger.h | 28 class StepStatsCollector; variable
77 StepStatsCollector* collector;
|
| D | graph_mgr.cc | 393 StepStatsCollector* collector, in ExecuteAsync()
471 StepStatsCollector* collector, in StartParallelExecutors()
520 void GraphMgr::BuildCostModel(Item* item, StepStatsCollector* collector, in BuildCostModel()
|
| D | worker_cache_logger.cc | 76 sl->collector = new StepStatsCollector(&sl->step_stats); in Save()
|
| D | worker.cc | 179 StepStatsCollector* collector = nullptr; in DoRunGraph()
183 collector = new StepStatsCollector(response->mutable_step_stats()); in DoRunGraph()
|
| /external/tensorflow/tensorflow/core/platform/ |
| D | device_tracer.h | 25 class StepStatsCollector; variable
70 virtual Status Collect(StepStatsCollector* collector) = 0;
|
| D | device_tracer_test.cc | 119 StepStatsCollector collector(&stats); in TEST_F()
129 StepStatsCollector collector(&stats); in TEST_F()
197 StepStatsCollector collector(&stats); in TEST_F()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.cc | 178 StepStatsCollector* step_stats_collector, in UpdateStats()
256 std::unique_ptr<StepStatsCollector> step_stats_collector; in Run()
258 step_stats_collector.reset(new StepStatsCollector(step_stats)); in Run()
329 std::unique_ptr<StepStatsCollector> step_stats_collector; in Run()
331 step_stats_collector.reset(new StepStatsCollector(step_stats)); in Run()
|
| /external/tensorflow/tensorflow/core/profiler/internal/cpu/ |
| D | host_tracer.cc | 61 absl::make_unique<StepStatsCollector>(run_metadata->mutable_step_stats()); in CollectData()
66 StepStatsCollector* step_stats_collector) { in CollectDataToCollector()
|
| D | host_tracer.h | 48 Status CollectDataToCollector(StepStatsCollector* step_stats_collector);
|
| /external/tensorflow/tensorflow/core/platform/default/ |
| D | device_tracer.cc | 401 Status Collect(StepStatsCollector* collector) override;
479 CudaEventCollector(CudaEventRecorder* recorder, StepStatsCollector* collector) in CudaEventCollector()
690 StepStatsCollector* collector) { in Collect()
700 StepStatsCollector* collector_;
709 Status DeviceTracerImpl::Collect(StepStatsCollector* collector) { in Collect()
|
| /external/tensorflow/tensorflow/core/profiler/internal/gpu/ |
| D | tracer.cc | 49 absl::make_unique<StepStatsCollector>(run_metadata->mutable_step_stats()); in CollectData()
|