Searched refs:StepStatsCollector (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | step_stats_collector.h | 38 class StepStatsCollector; variable
93 StepStatsCollector* step_stats_collector);
98 StepStatsCollector* step_stats_collector);
114 friend class StepStatsCollector;
130 StepStatsCollector* const step_stats_collector_; // Not owned.
156 class StepStatsCollector : public StepStatsCollectorInterface {
159 explicit StepStatsCollector(StepStats* step_stats);
|
| D | step_stats_collector.cc | 54 const NodeDef* node, StepStatsCollector* step_stats_collector) in NodeExecStatsWrapper()
62 StepStatsCollector* step_stats_collector) in NodeExecStatsWrapper()
196 StepStatsCollector::StepStatsCollector(StepStats* step_stats) in StepStatsCollector() function in tensorflow::StepStatsCollector
259 void StepStatsCollector::BuildCostModel( in BuildCostModel()
384 void StepStatsCollector::Save(const string& device, in Save()
391 void StepStatsCollector::Save(const string& device, in Save()
411 void StepStatsCollector::SaveThreadName(const string& device, in SaveThreadName()
426 NodeExecStatsInterface* StepStatsCollector::CreateNodeExecStats( in CreateNodeExecStats()
435 string StepStatsCollector::ReportAllocsOnResourceExhausted(const string& err) { in ReportAllocsOnResourceExhausted()
514 void StepStatsCollector::Finalize() { in Finalize()
[all …]
|
| D | local_executor_params.h | 25 class StepStatsCollector; variable
|
| D | executor.h | 34 class StepStatsCollector; variable
|
| D | direct_session.h | 203 std::unique_ptr<StepStatsCollector> collector;
|
| D | executor_test.cc | 100 StepStatsCollector step_stats_collector_;
|
| D | direct_session.cc | 672 new StepStatsCollector(run_metadata->mutable_step_stats())); in RunInternal()
1007 run_state->collector.reset(new StepStatsCollector(nullptr)); in PRunSetup()
|
| D | function_test.cc | 1347 StepStatsCollector stats_collector(&stats); in TEST_F()
1391 StepStatsCollector stats_collector(&stats); in TEST_F()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | graph_mgr.h | 43 class StepStatsCollector; variable
92 StepStatsCollector* collector,
169 StepStatsCollector* collector,
178 void BuildCostModel(Item* item, StepStatsCollector* collector,
|
| D | worker_cache_logger.h | 28 class StepStatsCollector; variable
77 StepStatsCollector* collector;
|
| D | worker_cache_logger.cc | 76 sl->collector = new StepStatsCollector(&sl->step_stats); in Save()
|
| D | graph_mgr.cc | 417 StepStatsCollector* collector, in ExecuteAsync()
513 CollectiveExecutor::Handle* ce_handle, StepStatsCollector* collector, in StartParallelExecutors()
561 void GraphMgr::BuildCostModel(Item* item, StepStatsCollector* collector, in BuildCostModel()
|
| D | worker.cc | 192 StepStatsCollector* collector = nullptr; in DoRunGraph()
196 collector = new StepStatsCollector(response->mutable_step_stats()); in DoRunGraph()
|
| /external/tensorflow/tensorflow/core/graph/ |
| D | costmodel_test.cc | 83 StepStatsCollector collector(&step_stats); in TEST()
|