Searched refs:step_stats_ (Results 1 – 5 of 5) sorted by relevance
197 : finalized_(false), step_stats_(step_stats) {} in StepStatsCollector()282 for (int i = 0; i < step_stats_->dev_stats_size(); ++i) { in BuildCostModel()283 const DeviceStepStats& device_stats = step_stats_->dev_stats(i); in BuildCostModel()400 if (!step_stats_ || collected_nodes_ >= kMaxCollectedNodes) { in Save()521 CHECK(step_stats_); in FinalizeAndSwap()523 step_stats->Swap(step_stats_); in FinalizeAndSwap()528 if (!step_stats_ || finalized_) { in FinalizeInternal()533 for (auto& ds : *step_stats_->mutable_dev_stats()) { in FinalizeInternal()538 DeviceStepStats* ndev_stat = step_stats_->add_dev_stats(); in FinalizeInternal()
201 StepStats* step_stats_ TF_GUARDED_BY(mu_);
57 step_stats_collector_(&step_stats_) {101 StepStats step_stats_; member in tensorflow::ExecutorTest
542 StepStats step_stats_; variable
636 return &step_stats_; in mutable_step_stats()