Searched refs:device_stats (Results 1 – 6 of 6) sorted by relevance
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
48 Stats* device_stats = &stats_[device]; in ReportAlloc() local49 device_stats->count += 1; in ReportAlloc()50 device_stats->size += msize; in ReportAlloc()51 return *device_stats; in ReportAlloc()56 Stats* device_stats = &stats_[device]; in ReportFree() local57 device_stats->count -= 1; in ReportFree()58 device_stats->size -= msize; in ReportFree()59 return *device_stats; in ReportFree()
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(data_generator(device_stats))
283 const DeviceStepStats& device_stats = step_stats_->dev_stats(i); in BuildCostModel() local284 const string& device_name = device_stats.device(); in BuildCostModel()288 gpu_hardware_stats.emplace(gpu_id, &device_stats); in BuildCostModel()292 DeviceStats{&device_stats, nullptr}); in BuildCostModel()405 auto& device_stats = dev_stats_[device]; in Save() local406 device_stats.push_back(std::unique_ptr<NodeExecStatsWrapper>(node_stats)); in Save()
414 for device_stats in self._step_stats.dev_stats:417 for ns in device_stats.node_stats:
126 device_stats = run_metadata.step_stats.dev_stats.add()127 device_stats.device = "test"