Home
last modified time | relevance | path

Searched refs:step_stats (Results 1 – 25 of 64) sorted by relevance

123

/external/tensorflow/tensorflow/python/client/
Dtimeline_test.py55 tl = timeline.Timeline(run_metadata.step_stats)
70 step_stats = run_metadata.step_stats
71 devices = [d.device for d in step_stats.dev_stats]
73 tl = timeline.Timeline(step_stats)
76 tl = timeline.Timeline(step_stats)
79 tl = timeline.Timeline(step_stats)
82 tl = timeline.Timeline(step_stats)
101 step_stats = run_metadata.step_stats
102 devices = [d.device for d in step_stats.dev_stats]
105 tl = timeline.Timeline(step_stats)
[all …]
Dvirtual_gpu_test.py228 step_stats = run_metadata.step_stats
229 devices = [d.device for d in step_stats.dev_stats]
Dsession_clusterspec_prop_test.py91 for dev_stats in run_metadata.step_stats.dev_stats
135 for dev_stats in run_metadata.step_stats.dev_stats
163 for dev_stats in run_metadata.step_stats.dev_stats
236 for dev_stats in run_metadata.step_stats.dev_stats
414 for dev_stats in run_metadata.step_stats.dev_stats
/external/tensorflow/tensorflow/core/distributed_runtime/
Dsession_mgr.cc150 auto step_stats = StepStats(); in RetrieveLogs() local
151 if (worker_cache->RetrieveLogs(step_id, &step_stats)) { in RetrieveLogs()
154 labeled_step_stats->mutable_step_stats()->Swap(&step_stats); in RetrieveLogs()
163 auto step_stats = StepStats(); in RetrieveLogs() local
164 if (worker_cache->RetrieveLogs(step_id, &step_stats)) { in RetrieveLogs()
167 labeled_step_stats->mutable_step_stats()->Swap(&step_stats); in RetrieveLogs()
Dworker_cache_logger.h76 StepStats step_stats; member
Dmessage_wrappers_test.cc158 ASSERT_EQ(1, response.metadata().step_stats().dev_stats_size()); in CheckRunStepResponse()
159 EXPECT_EQ("/cpu:0", response.metadata().step_stats().dev_stats(0).device()); in CheckRunStepResponse()
Dmaster_session.cc155 ss->MergeFrom(lss.step_stats()); in RetrieveLogs()
529 pss->step_stats.resize(partitions_.size()); in RunPartitions()
647 pss->step_stats[i].Swap(run_graph_resp->mutable_step_stats()); in RunPartitions()
749 const StepStats& ss = pss->step_stats[i]; in ProcessStats()
769 step_stats_proto.MergeFrom(pss->step_stats[i]); in ProcessStats()
770 pss->step_stats[i].Clear(); in ProcessStats()
772 pss->step_stats.clear(); in ProcessStats()
/external/tensorflow/tensorflow/python/grappler/
Dcluster_test.py45 op_perfs, run_time, step_stats = grappler_cluster.MeasureCosts(
49 self.assertTrue(step_stats.dev_stats)
62 op_perfs, run_time, step_stats = grappler_cluster.MeasureCosts(
66 self.assertEqual(len(step_stats.dev_stats), 0)
126 op_perfs, run_time, step_stats = gcluster.MeasureCosts(grappler_item)
129 self.assertTrue(step_stats.dev_stats)
Dcluster.i314 tensorflow::StepStats step_stats; in TF_MeasureCosts() local
330 step_stats = metadata.step_stats(); in TF_MeasureCosts()
353 string step_stats_str = step_stats.SerializeAsString(); in TF_MeasureCosts()
/external/tensorflow/tensorflow/python/util/
Dstat_summarizer.i63 tensorflow::StepStats step_stats; in ProcessStepStatsStr() local
64 step_stats.ParseFromString(step_stats_str); in ProcessStepStatsStr()
65 $self->ProcessStepStats(step_stats); in ProcessStepStatsStr()
/external/tensorflow/tensorflow/python/debug/cli/
Dprofile_analyzer_cli_test.py97 device1 = run_metadata.step_stats.dev_stats.add()
127 device1 = run_metadata.step_stats.dev_stats.add()
131 device2 = run_metadata.step_stats.dev_stats.add()
194 device1 = run_metadata.step_stats.dev_stats.add()
246 device1 = run_metadata.step_stats.dev_stats.add()
300 device1 = run_metadata.step_stats.dev_stats.add()
Dprofile_analyzer_cli.py417 device_count = len(self._run_metadata.step_stats.dev_stats)
419 device_stats = self._run_metadata.step_stats.dev_stats[index]
622 device_count = len(self._run_metadata.step_stats.dev_stats)
624 device_stats = self._run_metadata.step_stats.dev_stats[index]
/external/tensorflow/tensorflow/examples/tutorials/mnist/
Dmnist_softmax_xla.py81 trace = timeline.Timeline(step_stats=run_metadata.step_stats)
/external/tensorflow/tensorflow/python/eager/
Dcore_test.py95 step_stats = run_metadata.step_stats
96 self.assertGreater(len(step_stats.dev_stats), 0)
97 cpu_stats = step_stats.dev_stats[0]
/external/tensorflow/tensorflow/core/common_runtime/
Dstats_publisher_interface.cc29 void PublishStatsProto(const StepStats& step_stats) override {} in PublishStatsProto() argument
Dstats_publisher_interface.h35 virtual void PublishStatsProto(const StepStats& step_stats) = 0;
/external/tensorflow/tensorflow/core/grappler/costs/
Dutils.cc418 const auto& step_stats = run_metadata.step_stats(); in GetStatsStringFromRunMetadata() local
419 for (const auto& dev_stat : step_stats.dev_stats()) { in GetStatsStringFromRunMetadata()
Dgraph_memory.cc45 InferFromTrace(metadata.step_stats()); in InferStatically()
58 InferFromTrace(metadata.step_stats()); in InferDynamically()
/external/tensorflow/tensorflow/core/util/
Dstat_summarizer.cc126 void StatSummarizer::ProcessStepStats(const StepStats& step_stats) { in ProcessStepStats() argument
131 step_stats.dev_stats(0).node_stats(0).all_start_micros(); in ProcessStepStats()
134 for (const auto& ds : step_stats.dev_stats()) { in ProcessStepStats()
Dstat_summarizer_test.cc75 stats.ProcessStepStats(run_metadata.step_stats()); in TEST()
/external/tensorflow/tensorflow/examples/ios/benchmark/
DBenchmarkViewController.mm166 const tensorflow::StepStats& step_stats = run_metadata.step_stats();
167 stat_summarizer->ProcessStepStats(step_stats);
/external/tensorflow/tensorflow/python/profiler/
Dpprof_profiler.py284 self._run_metadata.step_stats.dev_stats):
293 device_count = len(self._run_metadata.step_stats.dev_stats)
/external/tensorflow/tensorflow/contrib/stat_summarizer/python/
Dstat_summarizer_test.py50 ss.ProcessStepStatsStr(run_metadata.step_stats.SerializeToString())
/external/tensorflow/tensorflow/contrib/android/jni/
Drun_stats_jni.cc66 s->ProcessStepStats(proto.step_stats()); in RUN_STATS_METHOD()
/external/tensorflow/tensorflow/tools/benchmark/
Dbenchmark_model.cc279 const StepStats& step_stats = run_metadata.step_stats(); in RunBenchmark() local
280 stats->ProcessStepStats(step_stats); in RunBenchmark()

123