Searched refs:device_stats (Results 1 – 7 of 7) sorted by relevance
42 auto* device_stats = in AssignLanes() local44 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() local81 device.set_name(device_stats->device()); in ConvertRunMetadataToTraceEvent()87 for (const auto& thread_name : device_stats->thread_names()) { in ConvertRunMetadataToTraceEvent()
54 Stats* device_stats = &stats_[device]; in ReportAlloc() local55 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() local63 device_stats->count -= 1; in ReportFree()64 device_stats->size -= msize; in ReportFree()65 return *device_stats; in ReportFree()
283 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
419 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)])
285 const DeviceStepStats& device_stats = step_stats_->dev_stats(i); in BuildCostModel() local286 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() local408 device_stats.push_back(std::unique_ptr<NodeExecStatsWrapper>(node_stats)); in Save()
401 for device_stats in self._step_stats.dev_stats:404 for ns in device_stats.node_stats:
125 device_stats = run_metadata.step_stats.dev_stats.add()126 device_stats.device = "test"