Home
last modified time | relevance | path

Searched refs:physical_device_desc (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device_factory.cc33 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()
Dgpu_device.h56 TfGpuId tf_gpu_id, const std::string& physical_device_desc,
363 const string& physical_device_desc, Allocator* gpu_allocator,
Dgpu_device.cc382 const string& physical_device_desc, in BaseGPUDevice() argument
387 physical_device_desc)), in BaseGPUDevice()
/external/tensorflow/tensorflow/core/framework/
Ddevice.cc44 const DeviceLocality& locality, const string& physical_device_desc) { in BuildDeviceAttributes() argument
53 da.set_physical_device_desc(physical_device_desc); in BuildDeviceAttributes()
Ddevice_attributes.proto52 string physical_device_desc = 7; field
Ddevice.h167 const DeviceLocality& locality, const std::string& physical_device_desc);
/external/tensorflow/tensorflow/core/common_runtime/
Ddevice_mgr.cc104 if (!dev->attributes().physical_device_desc().empty()) { in DeviceMappingString()
106 dev->attributes().physical_device_desc(), "\n"); in DeviceMappingString()
Ddynamic_device_mgr.cc71 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/
Dgpu_util.py52 device_attrs.physical_device_desc)
/external/tensorflow/tensorflow/python/eager/benchmarks/resnet50/
Dresnet50_test.py284 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/
Dsystem_info_lib.py140 device_info.physical_description = d.physical_device_desc
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Ddevice_util.cc52 if (r->match(device.attributes().physical_device_desc(), &cc)) { in ParseGpuDeviceMetadata()