| /external/tensorflow/tensorflow/core/common_runtime/device/ |
| D | device_id_manager.cc | 37 Status Insert(const DeviceType& type, TfDeviceId tf_device_id, in Insert()
62 bool Find(const DeviceType& type, TfDeviceId tf_device_id, in Find()
98 const DeviceType& type, TfDeviceId tf_device_id, in InsertTfPlatformDeviceIdPair()
105 const DeviceType& type, TfDeviceId tf_device_id, in TfToPlatformDeviceId()
|
| D | device_id_utils.h | 42 TfDeviceId tf_device_id) { in ExecutorForTfDeviceId()
53 TfDeviceId tf_device_id) { in CheckValidTfDeviceId()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_id_manager.cc | 24 TfDeviceId tf_device_id, PlatformDeviceId platform_device_id) { in InsertTfPlatformDeviceIdPair()
30 TfDeviceId tf_device_id, PlatformDeviceId* platform_device_id) { in TfToPlatformDeviceId()
|
| D | gpu_process_state.cc | 86 int GPUProcessState::BusIdForGPU(TfDeviceId tf_device_id) { in BusIdForGPU()
152 const GPUOptions& options, TfDeviceId tf_device_id, size_t total_bytes, in GetGPUAllocator()
263 SharedCounter* GPUProcessState::GPUAllocatorCounter(TfDeviceId tf_device_id) { in GPUAllocatorCounter()
|
| D | gpu_device_factory.cc | 33 TfDeviceId tf_device_id, const string& physical_device_desc, in GPUDevice()
66 const DeviceLocality& locality, TfDeviceId tf_device_id, in CreateGPUDevice()
|
| D | gpu_process_state.h | 88 Allocator* GetGPUAllocator(TfDeviceId tf_device_id) { in GetGPUAllocator()
|
| D | gpu_device.cc | 247 BaseGPUDevice::StreamGroup* GetOrCreate(TfDeviceId tf_device_id, in GetOrCreate()
350 int GetPriority(int tf_device_id, const GPUOptions& options) { in GetPriority()
389 TfDeviceId tf_device_id, in BaseGPUDevice()
904 TfDeviceId tf_device_id, in Reinitialize()
1518 TfDeviceId tf_device_id(di); in CreateDevices() local
1529 TfDeviceId tf_device_id(di); in CreateDevices() local
1561 TfDeviceId tf_device_id, int64_t memory_limit, in CreateGPUDevice()
1678 for (TfDeviceId tf_device_id : all_tf_device_ids) { in GetDeviceLocalities() local
|
| /external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| D | pluggable_device_factory.cc | 199 TfDeviceId tf_device_id(i); in CreateDevices() local
209 TfDeviceId tf_device_id(di); in CreateDevices() local
227 TfDeviceId tf_device_id, int64_t memory_limit, in CreatePluggableDevice()
284 TfDeviceId tf_device_id(i); in GetDeviceLocalities() local
|
| D | pluggable_device_process_state.cc | 68 TfDeviceId tf_device_id) { in BusIdForPluggableDevice()
80 const GPUOptions& options, TfDeviceId tf_device_id, size_t total_bytes) { in GetPluggableDeviceAllocator()
|
| D | pluggable_device.cc | 77 TfDeviceId tf_device_id, in GetOrCreate()
142 Bytes memory_limit, const DeviceLocality& locality, TfDeviceId tf_device_id, in PluggableDevice()
|
| /external/tensorflow/tensorflow/core/tfrt/eager/backends/gpu/ |
| D | gpu_registration.cc | 42 tensorflow::TfDeviceId tf_device_id(parsed_name.id); in RegisterGpuOpHandler() local
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | utils.cc | 159 TfDeviceId tf_device_id(device.id); in GetDeviceInfo() local
|
| D | single_machine.cc | 95 TfDeviceId tf_device_id(parsed.id); in Provision() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 112 TfDeviceId tf_device_id(tf_device_id_value); in GetFirstValidDeviceId() local
271 TfDeviceId tf_device_id; in GetEngineInfo() local
649 TfDeviceId tf_device_id; in GetDeviceAndAllocator() local
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | utils.cc | 251 TfDeviceId tf_device_id(parsed.id); in GetDeviceInfo() local
|
| /external/tensorflow/tensorflow/core/runtime_fallback/runtime/ |
| D | runtime_fallback_kernels.cc | 244 TfDeviceId tf_device_id{gpu_ordinal}; in InjectTfGpuResourcesHelper() local
|