Home
last modified time | relevance | path

Searched refs:device_stats (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/profiler/lib/
Dprofiler_session.cc42 auto* device_stats = in AssignLanes() local
44 if (device_stats->thread_names_size() > 0 || in AssignLanes()
45 device_stats->node_stats_size() == 0) { in AssignLanes()
49 for (auto ns = device_stats->mutable_node_stats()->rbegin(); in AssignLanes()
50 ns != device_stats->mutable_node_stats()->rend(); ns++) { in AssignLanes()
78 auto* device_stats = in ConvertRunMetadataToTraceEvent() local
81 device.set_name(device_stats->device()); in ConvertRunMetadataToTraceEvent()
87 for (const auto& thread_name : device_stats->thread_names()) { in ConvertRunMetadataToTraceEvent()
/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_state.cc54 Stats* device_stats = &stats_[device]; in ReportAlloc() local
55 device_stats->count += 1; in ReportAlloc()
56 device_stats->size += msize; in ReportAlloc()
57 return *device_stats; in ReportAlloc()
62 Stats* device_stats = &stats_[device]; in ReportFree() local
63 device_stats->count -= 1; in ReportFree()
64 device_stats->size -= msize; in ReportFree()
65 return *device_stats; in ReportFree()
/external/tensorflow/tensorflow/python/profiler/
Dpprof_profiler.py283 for device_index, device_stats in enumerate(
286 pprof_proto = self._get_pprof_proto(data_generator_func(device_stats))
290 'RunMetadata to session.run call?' % device_stats.device)
296 (device_index + 1, device_count, device_stats.device))
300 profiles[device_stats.device] = pprof_proto
/external/tensorflow/tensorflow/python/debug/cli/
Dprofile_analyzer_cli.py419 device_stats = self._run_metadata.step_stats.dev_stats[index]
420 if not device_name_regex or device_name_regex.match(device_stats.device):
422 datum for datum in data_generator(device_stats)
433 device_stats.device, index, device_count,
624 device_stats = self._run_metadata.step_stats.dev_stats[index]
625 if device_name_regex and not device_name_regex.match(device_stats.device):
627 profile_data.extend([datum for datum in data_generator(device_stats)])
/external/tensorflow/tensorflow/core/common_runtime/
Dstep_stats_collector.cc285 const DeviceStepStats& device_stats = step_stats_->dev_stats(i); in BuildCostModel() local
286 const string& device_name = device_stats.device(); in BuildCostModel()
290 gpu_hardware_stats.emplace(gpu_id, &device_stats); in BuildCostModel()
294 DeviceStats{&device_stats, nullptr}); in BuildCostModel()
407 auto& device_stats = dev_stats_[device]; in Save() local
408 device_stats.push_back(std::unique_ptr<NodeExecStatsWrapper>(node_stats)); in Save()
/external/tensorflow/tensorflow/python/client/
Dtimeline.py401 for device_stats in self._step_stats.dev_stats:
404 for ns in device_stats.node_stats:
/external/tensorflow/tensorflow/python/summary/writer/
Dwriter_test.py125 device_stats = run_metadata.step_stats.dev_stats.add()
126 device_stats.device = "test"