Home
last modified time | relevance | path

Searched refs:dev_stats (Results 1 – 25 of 42) sorted by relevance

12

/external/tensorflow/tensorflow/python/client/
Dtimeline.py414 for device_stats in self._step_stats.dev_stats:
487 for dev_stats in self._step_stats.dev_stats:
489 self._device_pids[dev_stats.device] = device_pid
491 self._tensor_pids[dev_stats.device] = tensors_pid
492 self._chrome_trace.emit_pid(dev_stats.device + ' Compute', device_pid)
493 self._chrome_trace.emit_pid(dev_stats.device + ' Tensors', tensors_pid)
497 for dev_stats in self._step_stats.dev_stats:
498 device_pid = self._device_pids[dev_stats.device]
499 tensors_pid = self._tensor_pids[dev_stats.device]
500 for node_stats in dev_stats.node_stats:
[all …]
Dsession_clusterspec_prop_test.py84 for dev_stats in run_metadata.step_stats.dev_stats
85 for node_stats in dev_stats.node_stats
87 dev_stats.device and 'Const' == node_stats.node_name
128 for dev_stats in run_metadata.step_stats.dev_stats
129 for node_stats in dev_stats.node_stats
131 dev_stats.device and 'Const' == node_stats.node_name
156 for dev_stats in run_metadata.step_stats.dev_stats
157 for node_stats in dev_stats.node_stats
159 == dev_stats.device and 'Const' == node_stats.node_name
229 for dev_stats in run_metadata.step_stats.dev_stats
[all …]
Dtimeline_test.py73 devices = [d.device for d in step_stats.dev_stats]
105 devices = [d.device for d in step_stats.dev_stats]
129 dev_stats = step_stats.dev_stats.add()
130 dev_stats.device = '/job:worker/replica:0/task:0/cpu:0'
131 node_stats = dev_stats.node_stats.add()
188 devices = [d.device for d in step_stats.dev_stats]
Dsession_test.py1476 self.assertEqual(0, len(run_metadata.step_stats.dev_stats))
1479 self.assertGreater(len(run_metadata.step_stats.dev_stats), 0)
1490 self.assertEqual(0, len(run_metadata.step_stats.dev_stats))
1493 self.assertGreater(len(run_metadata.step_stats.dev_stats), 0)
1504 self.assertEqual(0, len(run_metadata.step_stats.dev_stats))
1510 self.assertGreater(len(run_metadata.step_stats.dev_stats), 0)
1536 self.assertGreater(len(run_metadata.step_stats.dev_stats), 0)
1711 self.assertEqual(len(run_metadata.step_stats.dev_stats), 1)
1736 self.assertEqual(len(run_metadata.step_stats.dev_stats), 1)
Dvirtual_gpu_test.py229 devices = [d.device for d in step_stats.dev_stats]
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_memory.cc161 for (const auto& dev_stats : timeline.dev_stats()) { in InferFromTrace() local
162 for (const auto& node_stats : dev_stats.node_stats()) { in InferFromTrace()
163 node_placement[node_stats.node_name()] = dev_stats.device(); in InferFromTrace()
173 for (const auto& dev_stats : timeline.dev_stats()) { in InferFromTrace() local
174 const string& device_name = dev_stats.device(); in InferFromTrace()
177 live_tensors_per_device[dev_stats.device()]; in InferFromTrace()
178 for (const auto& node_stats : dev_stats.node_stats()) { in InferFromTrace()
Dmeasuring_cost_estimator.cc95 metadata.step_stats().dev_stats()) { in PredictCosts()
/external/tensorflow/tensorflow/core/profiler/internal/cpu/
Dhost_tracer.cc112 DeviceStepStats* dev_stats = step_stats->add_dev_stats(); in CollectData() local
113 dev_stats->set_device("/host:CPU"); in CollectData()
114 auto* thread_names = dev_stats->mutable_thread_names(); in CollectData()
123 NodeExecStats* ns = dev_stats->add_node_stats(); in CollectData()
Dhost_tracer_test.cc101 EXPECT_EQ(run_metadata.step_stats().dev_stats(0).node_stats_size(), 6); in TEST()
103 run_metadata.step_stats().dev_stats(0).node_stats(), in TEST()
/external/tensorflow/tensorflow/core/common_runtime/
Dstep_stats_collector.cc283 const DeviceStepStats& device_stats = step_stats_->dev_stats(i); in BuildCostModel()
323 const DeviceStats& dev_stats = per_device_stats.find(device)->second; in BuildCostModel() local
326 if (dev_stats.hardware_stats) { in BuildCostModel()
327 for (const auto& node_stats : dev_stats.hardware_stats->node_stats()) { in BuildCostModel()
348 for (int i = 0; i < dev_stats.regular_stats->node_stats_size(); ++i) { in BuildCostModel()
349 const NodeExecStats& stats = dev_stats.regular_stats->node_stats(i); in BuildCostModel()
369 string node_name = dev_stats.regular_stats->node_stats(i).node_name(); in BuildCostModel()
370 if (dev_stats.hardware_stats && name_to_hw_node_stats.find(node_name) != in BuildCostModel()
/external/tensorflow/tensorflow/compiler/tests/
Ddense_layer_test.py39 for dev_stats in run_metadata.step_stats.dev_stats:
40 for node_stats in dev_stats.node_stats:
Djit_test.py71 for dev_stats in run_metadata.step_stats.dev_stats:
72 for node_stats in dev_stats.node_stats:
/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.cc136 step_stats.dev_stats(0).node_stats_size() > 0) in ProcessStepStats()
137 ? step_stats.dev_stats(0).node_stats(0).all_start_micros() in ProcessStepStats()
141 for (const auto& ds : step_stats.dev_stats()) { in ProcessStepStats()
/external/tensorflow/tensorflow/python/grappler/
Dcluster_test.py49 self.assertTrue(step_stats.dev_stats)
66 self.assertEqual(len(step_stats.dev_stats), 0)
131 self.assertTrue(step_stats.dev_stats)
/external/tensorflow/tensorflow/core/profiler/internal/gpu/
Ddevice_tracer.cc209 for (auto& dev_stats : *step_stats.mutable_dev_stats()) { in CollectData()
210 run_metadata->mutable_step_stats()->add_dev_stats()->Swap(&dev_stats); in CollectData()
/external/tensorflow/tensorflow/python/kernel_tests/
Dtensor_array_ops_test.py1634 dev_stats = {d.device: d.node_stats
1635 for d in run_metadata.step_stats.dev_stats}
1636 for d in dev_stats:
1639 [s for s in dev_stats[d] if "/TensorArray" in s.node_name])
1642 [s for s in dev_stats[d] if "/TensorArray" in s.node_name])
1665 dev_stats = {d.device: d.node_stats
1666 for d in run_metadata.step_stats.dev_stats}
1667 for d in dev_stats:
1670 [s for s in dev_stats[d] if "TensorArray" == s.node_name])
1673 [s for s in dev_stats[d] if "TensorArray" == s.node_name])
[all …]
/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)
Dpprof_profiler_test.py75 device1 = run_metadata.step_stats.dev_stats.add()
Dtfprof_logger.py41 for dev_stat in run_meta.step_stats.dev_stats:
/external/tensorflow/tensorflow/core/distributed_runtime/
Dmessage_wrappers_test.cc124 EXPECT_EQ("/cpu:0", response->mutable_step_stats()->dev_stats(0).device()); in CheckRunGraphResponse()
159 EXPECT_EQ("/cpu:0", response.metadata().step_stats().dev_stats(0).device()); in CheckRunStepResponse()
/external/tensorflow/tensorflow/core/grappler/clusters/
Dvirtual_cluster_test.cc96 for (const auto& dev_stat : metadata.step_stats().dev_stats()) { in TEST_F()
/external/tensorflow/tensorflow/core/framework/
Dstep_stats.proto87 repeated DeviceStepStats dev_stats = 1; field
/external/tensorflow/tensorflow/python/profiler/internal/
Drun_metadata_test.py47 for dev_stat in run_meta.step_stats.dev_stats:

12