| /external/tensorflow/tensorflow/python/client/ |
| timeline_test.py | 54 self.assertTrue(run_metadata.HasField('step_stats'))
55 tl = timeline.Timeline(run_metadata.step_stats)
69 self.assertTrue(run_metadata.HasField('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)
[all...] |
| virtual_gpu_test.py | 227 self.assertTrue(run_metadata.HasField('step_stats'))
228 step_stats = run_metadata.step_stats
229 devices = [d.device for d in step_stats.dev_stats]
|
| session_clusterspec_prop_test.py | 91 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/python/util/ |
| stat_summarizer.i | 25 #include "tensorflow/core/framework/step_stats.pb.h"
63 tensorflow::StepStats step_stats;
64 step_stats.ParseFromString(step_stats_str);
65 $self->ProcessStepStats(step_stats);
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| stats_publisher_interface.cc | 29 void PublishStatsProto(const StepStats& step_stats) override {}
|
| stats_publisher_interface.h | 32 // PublishStatsProto publishes step_stats.
33 // When PublishStatsProto is called multiple times, only the step_stats
35 virtual void PublishStatsProto(const StepStats& step_stats) = 0;
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| session_mgr.cc | 150 auto step_stats = StepStats(); local
151 if (worker_cache->RetrieveLogs(step_id, &step_stats)) {
154 labeled_step_stats->mutable_step_stats()->Swap(&step_stats);
163 auto step_stats = StepStats(); local
164 if (worker_cache->RetrieveLogs(step_id, &step_stats)) {
167 labeled_step_stats->mutable_step_stats()->Swap(&step_stats);
|
| worker_cache_logger.h | 22 #include "tensorflow/core/framework/step_stats.pb.h"
76 StepStats step_stats; member in struct:tensorflow::WorkerCacheLogger::StepLog
|
| worker_cache_logger.cc | 76 sl->collector = new StepStatsCollector(&sl->step_stats);
|
| master_session.h | 152 std::vector<StepStats> step_stats; // per partition member in struct:tensorflow::MasterSession::PerStepState
|
| message_wrappers_test.cc | 19 #include "tensorflow/core/framework/step_stats.pb.h"
158 ASSERT_EQ(1, response.metadata().step_stats().dev_stats_size());
159 EXPECT_EQ("/cpu:0", response.metadata().step_stats().dev_stats(0).device());
|
| /external/tensorflow/tensorflow/python/grappler/ |
| cluster_test.py | 45 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)
|
| cluster.i | 314 tensorflow::StepStats step_stats;
330 step_stats = metadata.step_stats();
353 string step_stats_str = step_stats.SerializeAsString();
|
| /external/tensorflow/tensorflow/examples/tutorials/mnist/ |
| mnist_softmax_xla.py | 81 trace = timeline.Timeline(step_stats=run_metadata.step_stats)
|
| /external/tensorflow/tensorflow/core/protobuf/ |
| worker.proto | 26 import "tensorflow/core/framework/step_stats.proto";
248 StepStats step_stats = 2;
367 StepStats step_stats = 2;
|
| /external/tensorflow/tensorflow/core/util/ |
| stat_summarizer_test.cc | 19 #include "tensorflow/core/framework/step_stats.pb.h"
75 stats.ProcessStepStats(run_metadata.step_stats());
|
| stat_summarizer.h | 160 void ProcessStepStats(const StepStats& step_stats);
|
| /external/tensorflow/tensorflow/contrib/android/jni/ |
| run_stats_jni.cc | 66 s->ProcessStepStats(proto.step_stats());
|
| /external/tensorflow/tensorflow/contrib/stat_summarizer/python/ |
| stat_summarizer_test.py | 50 ss.ProcessStepStatsStr(run_metadata.step_stats.SerializeToString())
|
| /external/tensorflow/tensorflow/python/debug/cli/ |
| profile_analyzer_cli_test.py | 97 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()
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| virtual_cluster_test.cc | 20 #include "tensorflow/core/framework/step_stats.pb.h"
95 for (const auto& dev_stat : metadata.step_stats().dev_stats()) {
|
| /external/tensorflow/tensorflow/python/eager/ |
| core_test.py | 95 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/python/profiler/ |
| pprof_profiler_test.py | 74 device1 = run_metadata.step_stats.dev_stats.add()
|
| pprof_profiler.py | 284 self._run_metadata.step_stats.dev_stats):
293 device_count = len(self._run_metadata.step_stats.dev_stats)
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| graph_memory.cc | 21 #include "tensorflow/core/framework/step_stats.pb.h"
45 InferFromTrace(metadata.step_stats());
58 InferFromTrace(metadata.step_stats());
|
