Home
last modified time | relevance | path

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

123

/external/tensorflow/tensorflow/python/client/
Dtimeline_test.py56 tl = timeline.Timeline(run_metadata.step_stats)
72 step_stats = run_metadata.step_stats
73 devices = [d.device for d in step_stats.dev_stats]
75 tl = timeline.Timeline(step_stats)
78 tl = timeline.Timeline(step_stats)
81 tl = timeline.Timeline(step_stats)
84 tl = timeline.Timeline(step_stats)
104 step_stats = run_metadata.step_stats
105 devices = [d.device for d in step_stats.dev_stats]
108 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.py90 for dev_stats in run_metadata.step_stats.dev_stats
134 for dev_stats in run_metadata.step_stats.dev_stats
162 for dev_stats in run_metadata.step_stats.dev_stats
235 for dev_stats in run_metadata.step_stats.dev_stats
413 for dev_stats in run_metadata.step_stats.dev_stats
/external/tensorflow/tensorflow/core/distributed_runtime/
Dsession_mgr.cc174 auto step_stats = StepStats(); in RetrieveLogs() local
175 if (worker_cache->RetrieveLogs(step_id, &step_stats)) { in RetrieveLogs()
178 labeled_step_stats->mutable_step_stats()->Swap(&step_stats); in RetrieveLogs()
187 auto step_stats = StepStats(); in RetrieveLogs() local
188 if (worker_cache->RetrieveLogs(step_id, &step_stats)) { in RetrieveLogs()
191 labeled_step_stats->mutable_step_stats()->Swap(&step_stats); in RetrieveLogs()
Dworker_cache_logger.h76 StepStats step_stats; member
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.h83 StepStats* step_stats,
89 StepStats* step_stats,
149 StepStats* step_stats, GraphCollector* graph_collector) override;
154 StepStats* step_stats, GraphCollector* graph_collector) override;
211 StepStats* step_stats, GraphCollector* graph_collector) override;
215 StepStats* step_stats, GraphCollector* graph_collector) override;
Dkernel_and_device.cc153 NodeExecStats* stats, StepStats* step_stats, in Run() argument
158 return this->Run(&step_container, inputs, outputs, stats, step_stats, in Run()
164 std::vector<Tensor>* outputs, NodeExecStats* stats, StepStats* step_stats, in Run() argument
172 return this->Run(&step_container, inputs, outputs, stats, step_stats, in Run()
209 NodeExecStats* stats, StepStats* step_stats, in Run() argument
258 step_stats_collector.reset(new StepStatsCollector(step_stats)); in Run()
307 std::vector<Tensor>* outputs, NodeExecStats* stats, StepStats* step_stats, in Run() argument
331 step_stats_collector.reset(new StepStatsCollector(step_stats)); in Run()
Dexecute.cc57 void MaybeInitializeStepStats(StepStats* step_stats, EagerContext* ctx) { in MaybeInitializeStepStats() argument
60 while (step_stats->dev_stats_size() < ctx->devices()->size()) { in MaybeInitializeStepStats()
61 int device_idx = step_stats->dev_stats_size(); in MaybeInitializeStepStats()
62 auto* dev_stats = step_stats->add_dev_stats(); in MaybeInitializeStepStats()
67 int StepStatsDeviceIndex(StepStats* step_stats, EagerContext* ctx, in StepStatsDeviceIndex() argument
145 auto* step_stats = run_metadata->mutable_step_stats(); in MaybeCopyInputToExpectedDevice() local
146 MaybeInitializeStepStats(step_stats, ctx); in MaybeCopyInputToExpectedDevice()
148 int device_idx = StepStatsDeviceIndex(step_stats, ctx, handle_device); in MaybeCopyInputToExpectedDevice()
149 auto* dev_stats = step_stats->mutable_dev_stats(device_idx); in MaybeCopyInputToExpectedDevice()
987 auto* step_stats = ctx->RunMetadataProto()->mutable_step_stats(); in EagerKernelExecute() local
[all …]
/external/tensorflow/tensorflow/core/profiler/internal/cpu/
Dhost_tracer_test.cc91 EXPECT_EQ(run_metadata.step_stats().dev_stats_size(), 1); in TEST()
92 EXPECT_EQ(run_metadata.step_stats().dev_stats(0).node_stats_size(), 6); in TEST()
94 run_metadata.step_stats().dev_stats(0).node_stats(), in TEST()
110 run_metadata.step_stats().dev_stats(0).node_stats(), in ValidateResult()
/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)
128 op_perfs, run_time, step_stats = gcluster.MeasureCosts(grappler_item)
131 self.assertTrue(step_stats.dev_stats)
Dcluster.i321 tensorflow::StepStats step_stats; in TF_MeasureCosts() local
338 step_stats = metadata.step_stats(); in TF_MeasureCosts()
361 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/core/profiler/lib/
Dprofiler_session.cc41 device_id < run_metadata->step_stats().dev_stats_size(); ++device_id) { in AssignLanes()
76 device_id < run_metadata->step_stats().dev_stats_size(); ++device_id) { in ConvertRunMetadataToTraceEvent()
97 run_metadata->step_stats().dev_stats(device_id).node_stats()) { in ConvertRunMetadataToTraceEvent()
/external/tensorflow/tensorflow/python/debug/cli/
Dprofile_analyzer_cli_test.py98 device1 = run_metadata.step_stats.dev_stats.add()
128 device1 = run_metadata.step_stats.dev_stats.add()
132 device2 = run_metadata.step_stats.dev_stats.add()
195 device1 = run_metadata.step_stats.dev_stats.add()
247 device1 = run_metadata.step_stats.dev_stats.add()
301 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/core/util/
Dstat_summarizer.cc130 void StatSummarizer::ProcessStepStats(const StepStats& step_stats) { in ProcessStepStats() argument
135 step_stats.dev_stats(0).node_stats(0).all_start_micros(); in ProcessStepStats()
138 for (const auto& ds : step_stats.dev_stats()) { in ProcessStepStats()
Dstat_summarizer_test.cc74 stats.ProcessStepStats(run_metadata.step_stats()); in TEST()
Dstat_summarizer.h58 void ProcessStepStats(const StepStats& step_stats);
/external/tensorflow/tensorflow/examples/tutorials/mnist/
Dmnist_softmax_xla.py82 trace = timeline.Timeline(step_stats=run_metadata.step_stats)
/external/tensorflow/tensorflow/core/common_runtime/
Dstep_stats_collector.h164 explicit StepStatsCollector(StepStats* step_stats);
190 void FinalizeAndSwap(StepStats* step_stats);
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;
Dstep_stats_collector.cc198 StepStatsCollector::StepStatsCollector(StepStats* step_stats) in StepStatsCollector() argument
199 : finalized_(false), step_stats_(step_stats) {} in StepStatsCollector()
521 void StepStatsCollector::FinalizeAndSwap(StepStats* step_stats) { in FinalizeAndSwap() argument
525 step_stats->Swap(step_stats_); in FinalizeAndSwap()
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.-run-metadata.pbtxt6 name: "step_stats"
/external/tensorflow/tensorflow/core/grappler/costs/
Dutils.cc460 const auto& step_stats = run_metadata.step_stats(); in GetStatsStringFromRunMetadata() local
461 for (const auto& dev_stat : step_stats.dev_stats()) { in GetStatsStringFromRunMetadata()

123