Searched refs:physical_device_desc (Results 1 – 12 of 12) sorted by relevance
/external/tensorflow/tensorflow/core/common_runtime/gpu/ |
D | gpu_device_factory.cc | 33 TfGpuId tf_gpu_id, const string& physical_device_desc, in GPUDevice() argument 36 physical_device_desc, gpu_allocator, cpu_allocator, in GPUDevice() 67 const string& physical_device_desc, Allocator* gpu_allocator, in CreateGPUDevice() argument 70 tf_gpu_id, physical_device_desc, in CreateGPUDevice()
|
D | gpu_device.h | 56 TfGpuId tf_gpu_id, const std::string& physical_device_desc, 363 const string& physical_device_desc, Allocator* gpu_allocator,
|
D | gpu_device.cc | 382 const string& physical_device_desc, in BaseGPUDevice() argument 387 physical_device_desc)), in BaseGPUDevice()
|
/external/tensorflow/tensorflow/core/framework/ |
D | device.cc | 44 const DeviceLocality& locality, const string& physical_device_desc) { in BuildDeviceAttributes() argument 53 da.set_physical_device_desc(physical_device_desc); in BuildDeviceAttributes()
|
D | device_attributes.proto | 52 string physical_device_desc = 7; field
|
D | device.h | 167 const DeviceLocality& locality, const std::string& physical_device_desc);
|
/external/tensorflow/tensorflow/core/common_runtime/ |
D | device_mgr.cc | 104 if (!dev->attributes().physical_device_desc().empty()) { in DeviceMappingString() 106 dev->attributes().physical_device_desc(), "\n"); in DeviceMappingString()
|
D | dynamic_device_mgr.cc | 71 if (!pair.first->attributes().physical_device_desc().empty()) { in DeviceMappingString() 73 pair.first->attributes().physical_device_desc(), "\n"); in DeviceMappingString()
|
/external/tensorflow/tensorflow/python/framework/ |
D | gpu_util.py | 52 device_attrs.physical_device_desc)
|
/external/tensorflow/tensorflow/python/eager/benchmarks/resnet50/ |
D | resnet50_test.py | 284 if 'K20' in device.physical_device_desc: 287 if 'P1000' in device.physical_device_desc: 289 if 'P100' in device.physical_device_desc:
|
/external/tensorflow/tensorflow/tools/test/ |
D | system_info_lib.py | 140 device_info.physical_description = d.physical_device_desc
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
D | device_util.cc | 52 if (r->match(device.attributes().physical_device_desc(), &cc)) { in ParseGpuDeviceMetadata()
|