Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/client/
Dsession_clusterspec_prop_test.py91 for dev_stats in run_metadata.step_stats.dev_stats
92 for node_stats in dev_stats.node_stats
94 dev_stats.device and 'Const' == node_stats.node_name
135 for dev_stats in run_metadata.step_stats.dev_stats
136 for node_stats in dev_stats.node_stats
138 dev_stats.device and 'Const' == node_stats.node_name
163 for dev_stats in run_metadata.step_stats.dev_stats
164 for node_stats in dev_stats.node_stats
166 == dev_stats.device and 'Const' == node_stats.node_name
236 for dev_stats in run_metadata.step_stats.dev_stats
[all …]
Dtimeline.py401 for device_stats in self._step_stats.dev_stats:
475 for dev_stats in self._step_stats.dev_stats:
477 self._device_pids[dev_stats.device] = device_pid
479 self._tensor_pids[dev_stats.device] = tensors_pid
480 self._chrome_trace.emit_pid(dev_stats.device + ' Compute', device_pid)
481 self._chrome_trace.emit_pid(dev_stats.device + ' Tensors', tensors_pid)
485 for dev_stats in self._step_stats.dev_stats:
486 device_pid = self._device_pids[dev_stats.device]
487 tensors_pid = self._tensor_pids[dev_stats.device]
488 for node_stats in dev_stats.node_stats:
[all …]
Dtimeline_test.py71 devices = [d.device for d in step_stats.dev_stats]
102 devices = [d.device for d in step_stats.dev_stats]
123 dev_stats = step_stats.dev_stats.add()
124 dev_stats.device = '/job:worker/replica:0/task:0/cpu:0'
125 node_stats = dev_stats.node_stats.add()
183 devices = [d.device for d in step_stats.dev_stats]
Dsession_test.py1270 self.assertEqual(0, len(run_metadata.step_stats.dev_stats))
1273 self.assertGreater(len(run_metadata.step_stats.dev_stats), 0)
1284 self.assertEqual(0, len(run_metadata.step_stats.dev_stats))
1287 self.assertGreater(len(run_metadata.step_stats.dev_stats), 0)
1298 self.assertEqual(0, len(run_metadata.step_stats.dev_stats))
1304 self.assertGreater(len(run_metadata.step_stats.dev_stats), 0)
1479 self.assertEquals(len(run_metadata.step_stats.dev_stats), 1)
1504 self.assertEquals(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.cc156 for (const auto& dev_stats : timeline.dev_stats()) { in InferFromTrace() local
157 for (const auto& node_stats : dev_stats.node_stats()) { in InferFromTrace()
158 node_placement[node_stats.node_name()] = dev_stats.device(); in InferFromTrace()
166 for (const auto& dev_stats : timeline.dev_stats()) { in InferFromTrace() local
167 const string& device_name = dev_stats.device(); in InferFromTrace()
170 live_tensors_per_device[dev_stats.device()]; in InferFromTrace()
171 for (const auto& node_stats : dev_stats.node_stats()) { in InferFromTrace()
Dvirtual_scheduler_test.cc1574 EXPECT_EQ(1, stepstats.dev_stats().size()); in TEST_F()
1578 for (const auto& device_step_stats : stepstats.dev_stats()) { in TEST_F()
1844 for (const auto& device_step_stats : metadata.step_stats().dev_stats()) { in TEST_F()
/external/tensorflow/tensorflow/core/common_runtime/
Dstep_stats_collector.cc158 const DeviceStepStats& device_stats = step_stats_->dev_stats(i); in BuildCostModel()
198 const DeviceStats& dev_stats = per_device_stats.find(device)->second; in BuildCostModel() local
201 if (dev_stats.hardware_stats) { in BuildCostModel()
202 for (const auto& node_stats : dev_stats.hardware_stats->node_stats()) { in BuildCostModel()
223 for (int i = 0; i < dev_stats.regular_stats->node_stats_size(); ++i) { in BuildCostModel()
224 const NodeExecStats& stats = dev_stats.regular_stats->node_stats(i); in BuildCostModel()
243 string node_name = dev_stats.regular_stats->node_stats(i).node_name(); in BuildCostModel()
244 if (dev_stats.hardware_stats && name_to_hw_node_stats.find(node_name) != in BuildCostModel()
/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/python/kernel_tests/
Dtensor_array_ops_test.py1356 dev_stats = {d.device: d.node_stats
1357 for d in run_metadata.step_stats.dev_stats}
1358 for d in dev_stats:
1361 [s for s in dev_stats[d] if "/TensorArray" in s.node_name])
1364 [s for s in dev_stats[d] if "/TensorArray" in s.node_name])
1385 dev_stats = {d.device: d.node_stats
1386 for d in run_metadata.step_stats.dev_stats}
1387 for d in dev_stats:
1390 [s for s in dev_stats[d] if "/TensorArray" in s.node_name])
1393 [s for s in dev_stats[d] if "/TensorArray" in s.node_name])
[all …]
/external/tensorflow/tensorflow/python/grappler/
Dcluster_test.py49 self.assertTrue(step_stats.dev_stats)
66 self.assertEqual(len(step_stats.dev_stats), 0)
129 self.assertTrue(step_stats.dev_stats)
/external/tensorflow/tensorflow/compiler/tests/
Djit_test.py55 for dev_stats in run_metadata.step_stats.dev_stats:
56 for node_stats in dev_stats.node_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)
Dpprof_profiler_test.py74 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.cc95 for (const auto& dev_stat : metadata.step_stats().dev_stats()) { in TEST_F()
/external/tensorflow/tensorflow/core/framework/
Dstep_stats.proto78 repeated DeviceStepStats dev_stats = 1; field
/external/tensorflow/tensorflow/core/util/
Dstat_summarizer.cc131 step_stats.dev_stats(0).node_stats(0).all_start_micros(); in ProcessStepStats()
134 for (const auto& ds : step_stats.dev_stats()) { in ProcessStepStats()
/external/tensorflow/tensorflow/python/eager/
Dcore_test.py96 self.assertGreater(len(step_stats.dev_stats), 0)
97 cpu_stats = step_stats.dev_stats[0]
/external/tensorflow/tensorflow/python/profiler/internal/
Drun_metadata_test.py45 for dev_stat in run_meta.step_stats.dev_stats:
/external/tensorflow/tensorflow/c/eager/
Dc_api.cc874 auto* dev_stats = step_stats->mutable_dev_stats(device_idx); in TFE_Execute() local
875 dev_stats->set_device(device->name()); in TFE_Execute()
876 *dev_stats->add_node_stats() = *maybe_stats; in TFE_Execute()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dmemory_optimizer.cc988 for (const auto& dev_stats : metadata.step_stats().dev_stats()) { in IdentifySwappingCandidates() local
989 for (const auto& node_stats : dev_stats.node_stats()) { in IdentifySwappingCandidates()
/external/tensorflow/tensorflow/core/profiler/internal/
Dtfprof_stats.cc276 for (const auto& dev_stat : run_meta->step_stats().dev_stats()) { in AddRunMeta()

12