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.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 …]
Dsession_clusterspec_prop_test.py90 for dev_stats in run_metadata.step_stats.dev_stats
91 for node_stats in dev_stats.node_stats
93 dev_stats.device and 'Const' == node_stats.node_name
134 for dev_stats in run_metadata.step_stats.dev_stats
135 for node_stats in dev_stats.node_stats
137 dev_stats.device and 'Const' == node_stats.node_name
162 for dev_stats in run_metadata.step_stats.dev_stats
163 for node_stats in dev_stats.node_stats
165 == dev_stats.device and 'Const' == node_stats.node_name
235 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]
126 dev_stats = step_stats.dev_stats.add()
127 dev_stats.device = '/job:worker/replica:0/task:0/cpu:0'
128 node_stats = dev_stats.node_stats.add()
185 devices = [d.device for d in step_stats.dev_stats]
Dsession_test.py1456 self.assertEqual(0, len(run_metadata.step_stats.dev_stats))
1459 self.assertGreater(len(run_metadata.step_stats.dev_stats), 0)
1470 self.assertEqual(0, len(run_metadata.step_stats.dev_stats))
1473 self.assertGreater(len(run_metadata.step_stats.dev_stats), 0)
1484 self.assertEqual(0, len(run_metadata.step_stats.dev_stats))
1490 self.assertGreater(len(run_metadata.step_stats.dev_stats), 0)
1516 self.assertGreater(len(run_metadata.step_stats.dev_stats), 0)
1691 self.assertEquals(len(run_metadata.step_stats.dev_stats), 1)
1716 self.assertEquals(len(run_metadata.step_stats.dev_stats), 1)
/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_memory.cc162 for (const auto& dev_stats : timeline.dev_stats()) { in InferFromTrace() local
163 for (const auto& node_stats : dev_stats.node_stats()) { in InferFromTrace()
164 node_placement[node_stats.node_name()] = dev_stats.device(); in InferFromTrace()
174 for (const auto& dev_stats : timeline.dev_stats()) { in InferFromTrace() local
175 const string& device_name = dev_stats.device(); in InferFromTrace()
178 live_tensors_per_device[dev_stats.device()]; in InferFromTrace()
179 for (const auto& node_stats : dev_stats.node_stats()) { in InferFromTrace()
Dmeasuring_cost_estimator.cc95 metadata.step_stats().dev_stats()) { in PredictCosts()
Dvirtual_scheduler_test.cc2313 EXPECT_EQ(1, stepstats.dev_stats().size()); in TEST_F()
2317 for (const auto& device_step_stats : stepstats.dev_stats()) { in TEST_F()
2561 for (const auto& device_step_stats : metadata.step_stats().dev_stats()) { in TEST_F()
2673 for (const auto& device_step_stats : metadata.step_stats().dev_stats()) { in TEST_F()
2735 for (const auto& device_step_stats : metadata.step_stats().dev_stats()) { in TEST_F()
/external/tensorflow/tensorflow/core/profiler/internal/cpu/
Dhost_tracer_test.cc92 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/core/common_runtime/
Dstep_stats_collector.cc285 const DeviceStepStats& device_stats = step_stats_->dev_stats(i); in BuildCostModel()
325 const DeviceStats& dev_stats = per_device_stats.find(device)->second; in BuildCostModel() local
328 if (dev_stats.hardware_stats) { in BuildCostModel()
329 for (const auto& node_stats : dev_stats.hardware_stats->node_stats()) { in BuildCostModel()
350 for (int i = 0; i < dev_stats.regular_stats->node_stats_size(); ++i) { in BuildCostModel()
351 const NodeExecStats& stats = dev_stats.regular_stats->node_stats(i); in BuildCostModel()
371 string node_name = dev_stats.regular_stats->node_stats(i).node_name(); in BuildCostModel()
372 if (dev_stats.hardware_stats && name_to_hw_node_stats.find(node_name) != in BuildCostModel()
/external/tensorflow/tensorflow/compiler/tests/
Ddense_layer_test.py37 for dev_stats in run_metadata.step_stats.dev_stats:
38 for node_stats in dev_stats.node_stats:
Djit_test.py70 for dev_stats in run_metadata.step_stats.dev_stats:
71 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/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/python/kernel_tests/
Dtensor_array_ops_test.py1523 dev_stats = {d.device: d.node_stats
1524 for d in run_metadata.step_stats.dev_stats}
1525 for d in dev_stats:
1528 [s for s in dev_stats[d] if "/TensorArray" in s.node_name])
1531 [s for s in dev_stats[d] if "/TensorArray" in s.node_name])
1554 dev_stats = {d.device: d.node_stats
1555 for d in run_metadata.step_stats.dev_stats}
1556 for d in dev_stats:
1559 [s for s in dev_stats[d] if "TensorArray" == s.node_name])
1562 [s for s in dev_stats[d] if "TensorArray" == s.node_name])
[all …]
/external/tensorflow/tensorflow/core/util/
Dstat_summarizer.cc135 step_stats.dev_stats(0).node_stats(0).all_start_micros(); in ProcessStepStats()
138 for (const auto& ds : step_stats.dev_stats()) { in ProcessStepStats()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dexecute.cc62 auto* dev_stats = step_stats->add_dev_stats(); in MaybeInitializeStepStats() local
63 dev_stats->set_device(ctx->devices()->at(device_idx)->name()); in MaybeInitializeStepStats()
149 auto* dev_stats = step_stats->mutable_dev_stats(device_idx); in MaybeCopyInputToExpectedDevice() local
150 auto* node_stats = dev_stats->add_node_stats(); in MaybeCopyInputToExpectedDevice()
1008 auto* dev_stats = step_stats->mutable_dev_stats(device_idx); in EagerKernelExecute() local
1009 dev_stats->set_device(attribution_device->name()); in EagerKernelExecute()
1010 *dev_stats->add_node_stats() = *maybe_stats; in EagerKernelExecute()
/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.cc95 for (const auto& dev_stat : metadata.step_stats().dev_stats()) { in TEST_F()
/external/tensorflow/tensorflow/core/framework/
Dstep_stats.proto85 repeated DeviceStepStats dev_stats = 1; field
/external/tensorflow/tensorflow/core/profiler/lib/
Dprofiler_session.cc97 run_metadata->step_stats().dev_stats(device_id).node_stats()) { in ConvertRunMetadataToTraceEvent()
/external/tensorflow/tensorflow/python/profiler/internal/
Drun_metadata_test.py47 for dev_stat in run_meta.step_stats.dev_stats:

12