| /external/tensorflow/tensorflow/stream_executor/ |
| D | device_description.cc | 27 /* static */ const char *DeviceDescription::kUndefinedString = "<undefined>";
29 DeviceDescription::DeviceDescription() in DeviceDescription() function in stream_executor::DeviceDescription
59 std::unique_ptr<std::map<string, string>> DeviceDescription::ToMap() const { in ToMap()
107 : device_description_(new DeviceDescription) {} in DeviceDescriptionBuilder()
111 bool DeviceDescription::cuda_compute_capability(int *major, int *minor) const { in cuda_compute_capability()
117 bool DeviceDescription::rocm_amdgpu_isa_version(int *version) const { in rocm_amdgpu_isa_version()
126 bool ThreadDimOk(const DeviceDescription &device_description, in ThreadDimOk()
150 void CalculateDimensionality(const DeviceDescription &device_description, in CalculateDimensionality()
|
| D | device_description.h | 40 class DeviceDescription {
170 DeviceDescription();
216 SE_DISALLOW_COPY_AND_ASSIGN(DeviceDescription);
316 std::unique_ptr<DeviceDescription> Build() { in Build()
321 std::unique_ptr<DeviceDescription> device_description_;
331 bool ThreadDimOk(const DeviceDescription &device_description,
342 void CalculateDimensionality(const DeviceDescription &device_description,
|
| D | stream_executor_pimpl.h | 347 const DeviceDescription &GetDeviceDescription() const;
611 DeviceDescription *PopulateDeviceDescription() const;
674 mutable std::unique_ptr<DeviceDescription> device_description_
|
| D | stream_executor_internal.h | 291 virtual DeviceDescription *PopulateDeviceDescription() const = 0;
|
| D | stream_executor_pimpl.cc | 293 const DeviceDescription &StreamExecutor::GetDeviceDescription() const { in GetDeviceDescription()
812 DeviceDescription *StreamExecutor::PopulateDeviceDescription() const { in PopulateDeviceDescription()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_context.h | 35 const se::DeviceDescription* device_desc, in IrEmitterContext()
50 const se::DeviceDescription& device_description() const { in device_description()
59 const se::DeviceDescription* device_desc_;
|
| D | partition_assignment.h | 61 int64 ThreadsPerBlockLimit(const se::DeviceDescription& device_desc);
64 const Shape& shape, const se::DeviceDescription& device_desc,
|
| D | partition_assignment.cc | 42 int64 ThreadsPerBlockLimit(const se::DeviceDescription& device_desc) { in ThreadsPerBlockLimit()
63 const Shape& shape, const se::DeviceDescription& device_desc, in CalculateLaunchDimensions()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_execution_profile.h | 108 using DeviceDescription = se::DeviceDescription;
144 string ToString(const DeviceDescription& device_description) const { in ToString()
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_executor.h | 77 int CalculateOccupancy(const DeviceDescription& device_description,
84 const DeviceDescription& device_description,
201 DeviceDescription* PopulateDeviceDescription() const override;
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | devices.cc | 41 const se::DeviceDescription& desc = se->GetDeviceDescription(); in GetNumAvailableGPUs()
|
| /external/autotest/client/common_lib/cros/network/ |
| D | interface.py | 17 DeviceDescription = collections.namedtuple('DeviceDescription', variable
263 return DeviceDescription(device_name, module_path)
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executor.cc | 115 DeviceDescription *XlaInterpreterExecutor::PopulateDeviceDescription() const { in PopulateDeviceDescription()
|
| D | executor.h | 168 DeviceDescription *PopulateDeviceDescription() const override;
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_gpu_executor.cc | 484 const DeviceDescription &device_description = in VlogOccupancyInfo()
508 int GpuExecutor::CalculateOccupancy(const DeviceDescription& device_description, in CalculateOccupancy()
525 const DeviceDescription& device_description, in CompareOccupancy()
1028 DeviceDescription* GpuExecutor::PopulateDeviceDescription() const { in PopulateDeviceDescription()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_gpu_executor.cc | 344 int GpuExecutor::CalculateOccupancy(const DeviceDescription& device_description, in CalculateOccupancy()
354 const DeviceDescription& device_description, in CompareOccupancy()
875 DeviceDescription* GpuExecutor::PopulateDeviceDescription() const { in PopulateDeviceDescription()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 1162 const se::DeviceDescription& desc) { in GetShortDeviceDescription()
1197 const se::DeviceDescription& desc = se->GetDeviceDescription(); in CreateGPUDevice()
1292 const se::DeviceDescription& desc = se->GetDeviceDescription(); in GetDeviceLocalities()
1367 const se::DeviceDescription& desc = se->GetDeviceDescription(); in GetDefaultMinGPUMultiprocessorCount()
1613 const se::DeviceDescription& desc = se->GetDeviceDescription(); in GetValidDeviceIds()
|
| /external/tensorflow/tensorflow/stream_executor/host/ |
| D | host_gpu_executor.h | 150 DeviceDescription *PopulateDeviceDescription() const override;
|
| D | host_gpu_executor.cc | 187 DeviceDescription *HostExecutor::PopulateDeviceDescription() const { in PopulateDeviceDescription()
|