Searched refs:step_stats_ (Results 1 – 5 of 5) sorted by relevance
199 : finalized_(false), step_stats_(step_stats) {} in StepStatsCollector()284 for (int i = 0; i < step_stats_->dev_stats_size(); ++i) { in BuildCostModel()285 const DeviceStepStats& device_stats = step_stats_->dev_stats(i); in BuildCostModel()402 if (!step_stats_ || collected_nodes_ >= kMaxCollectedNodes) { in Save()523 CHECK(step_stats_); in FinalizeAndSwap()525 step_stats->Swap(step_stats_); in FinalizeAndSwap()530 if (!step_stats_ || finalized_) { in FinalizeInternal()535 for (auto& ds : *step_stats_->mutable_dev_stats()) { in FinalizeInternal()540 DeviceStepStats* ndev_stat = step_stats_->add_dev_stats(); in FinalizeInternal()
206 StepStats* step_stats_ GUARDED_BY(mu_);
46 step_stats_collector_(&step_stats_) {89 StepStats step_stats_; member in tensorflow::ExecutorTest
530 StepStats step_stats_; variable
617 return &step_stats_; in mutable_step_stats()