Home
last modified time | relevance | path

Searched defs:step_stats (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/core/distributed_runtime/
Dworker_cache_logger.h76 StepStats step_stats; member
Dsession_mgr.cc174 auto step_stats = StepStats(); in RetrieveLogs() local
187 auto step_stats = StepStats(); in RetrieveLogs() local
Dmaster_session.h170 std::vector<StepStats> step_stats; // per partition member
/external/tensorflow/tensorflow/python/util/
Dstat_summarizer.i63 tensorflow::StepStats step_stats; in ProcessStepStatsStr() local
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.cc153 NodeExecStats* stats, StepStats* step_stats, in Run()
164 std::vector<Tensor>* outputs, NodeExecStats* stats, StepStats* step_stats, in Run()
209 NodeExecStats* stats, StepStats* step_stats, in Run()
307 std::vector<Tensor>* outputs, NodeExecStats* stats, StepStats* step_stats, in Run()
Dexecute.cc57 void MaybeInitializeStepStats(StepStats* step_stats, EagerContext* ctx) { in MaybeInitializeStepStats()
67 int StepStatsDeviceIndex(StepStats* step_stats, EagerContext* ctx, in StepStatsDeviceIndex()
145 auto* step_stats = run_metadata->mutable_step_stats(); in MaybeCopyInputToExpectedDevice() local
987 auto* step_stats = ctx->RunMetadataProto()->mutable_step_stats(); in EagerKernelExecute() local
/external/tensorflow/tensorflow/core/common_runtime/
Dstats_publisher_interface.cc29 void PublishStatsProto(const StepStats& step_stats) override {} in PublishStatsProto()
Dstep_stats_collector.cc198 StepStatsCollector::StepStatsCollector(StepStats* step_stats) in StepStatsCollector()
521 void StepStatsCollector::FinalizeAndSwap(StepStats* step_stats) { in FinalizeAndSwap()
/external/tensorflow/tensorflow/core/util/
Dstat_summarizer.cc130 void StatSummarizer::ProcessStepStats(const StepStats& step_stats) { in ProcessStepStats()
/external/tensorflow/tensorflow/core/protobuf/
Dworker.proto267 StepStats step_stats = 2; field
389 StepStats step_stats = 2; field
Dconfig.proto516 StepStats step_stats = 1; field
/external/tensorflow/tensorflow/python/grappler/
Dcluster.i321 tensorflow::StepStats step_stats; in TF_MeasureCosts() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dutils.cc460 const auto& step_stats = run_metadata.step_stats(); in GetStatsStringFromRunMetadata() local
/external/tensorflow/tensorflow/python/client/
Dtimeline.py349 def __init__(self, step_stats, graph=None): argument
/external/tensorflow/tensorflow/tools/benchmark/
Dbenchmark_model.cc313 const StepStats& step_stats = run_metadata.step_stats(); in RunBenchmark() local
/external/tensorflow/tensorflow/python/training/
Dbasic_session_run_hooks.py1055 def _save(self, step, save_path, step_stats): argument