Home
last modified time | relevance | path

Searched defs:platform_device_id (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/core/common_runtime/device/
Ddevice_id_utils.h37 se::Platform* device_manager, PlatformDeviceId platform_device_id) { in ExecutorForPlatformDeviceId()
43 PlatformDeviceId platform_device_id; in ExecutorForTfDeviceId() local
54 PlatformDeviceId platform_device_id; in CheckValidTfDeviceId() local
81 int32_t platform_device_id; in ParseVisibleDeviceList() local
Ddevice_id_manager.cc38 PlatformDeviceId platform_device_id) TF_LOCKS_EXCLUDED(mu_) { in Insert()
99 PlatformDeviceId platform_device_id) { in InsertTfPlatformDeviceIdPair()
106 PlatformDeviceId* platform_device_id) { in TfToPlatformDeviceId()
Ddevice_id_manager_test.cc27 PlatformDeviceId platform_device_id; in TfToPlatformDeviceId() local
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_debug_allocator_test.cc41 PlatformDeviceId platform_device_id) { in ExecutorForPlatformDeviceId()
48 const PlatformDeviceId platform_device_id(0); in TEST() local
150 const PlatformDeviceId platform_device_id(0); in TEST() local
194 const PlatformDeviceId platform_device_id(0); in TEST() local
239 const PlatformDeviceId platform_device_id(0); in TEST() local
251 const PlatformDeviceId platform_device_id(0); in TEST() local
Dgpu_id_manager.cc24 TfDeviceId tf_device_id, PlatformDeviceId platform_device_id) { in InsertTfPlatformDeviceIdPair()
30 TfDeviceId tf_device_id, PlatformDeviceId* platform_device_id) { in TfToPlatformDeviceId()
Dgpu_device.cc129 PlatformDeviceId platform_device_id, in Reinitialize()
495 PlatformDeviceId platform_device_id; in Init() local
907 PlatformDeviceId platform_device_id; in Reinitialize() local
917 PlatformDeviceId platform_device_id, in Reinitialize()
1086 PlatformDeviceId platform_device_id, in SingleVirtualDeviceMemoryLimit()
1231 for (PlatformDeviceId platform_device_id : cached_device_ids_) { in ListPhysicalDevices() local
1247 PlatformDeviceId platform_device_id = cached_device_ids_[device_index]; in GetDeviceDetails() local
1342 for (PlatformDeviceId platform_device_id : valid_platform_device_ids) { in CreateDevices() local
1455 PlatformDeviceId platform_device_id; in CreateDevices() member
1472 const PlatformDeviceId platform_device_id = valid_platform_device_ids[i]; in CreateDevices() local
[all …]
Dgpu_cudamalloc_allocator.cc30 PlatformDeviceId platform_device_id) { in GPUcudaMallocAllocator()
Dgpu_process_state.cc99 const GPUOptions& options, PlatformDeviceId platform_device_id, in CreateSubAllocator()
174 PlatformDeviceId platform_device_id; in GetGPUAllocator() local
Dgpu_debug_allocator.cc79 PlatformDeviceId platform_device_id) in GPUDebugAllocator()
158 PlatformDeviceId platform_device_id) in GPUNanResetAllocator()
Dgpu_cudamallocasync_allocator.cc101 PlatformDeviceId platform_device_id, size_t pool_size, bool reserve_memory, in GpuCudaMallocAsyncAllocator()
Dgpu_device.h114 PlatformDeviceId platform_device_id; in gpu_id() local
/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/
Dpluggable_device_factory.cc80 PlatformDeviceId platform_device_id, in SingleVirtualDeviceMemoryLimit()
193 const PlatformDeviceId platform_device_id = visible_device_order[i]; in CreateDevices() local
218 static string GetShortDeviceDescription(PlatformDeviceId platform_device_id, in GetShortDeviceDescription()
237 PlatformDeviceId platform_device_id; in CreatePluggableDevice() local
285 PlatformDeviceId platform_device_id; in GetDeviceLocalities() local
Dpluggable_device_process_state.cc101 PlatformDeviceId platform_device_id; in GetPluggableDeviceAllocator() local
Dpluggable_device.cc187 PlatformDeviceId platform_device_id; in Init() local
/external/tensorflow/tensorflow/core/grappler/clusters/
Dutils.cc77 DeviceProperties GetLocalGPUInfo(PlatformDeviceId platform_device_id) { in GetLocalGPUInfo()
160 PlatformDeviceId platform_device_id; in GetDeviceInfo() local
Dsingle_machine.cc96 PlatformDeviceId platform_device_id; in Provision() local
/external/tensorflow/tensorflow/core/tfrt/eager/backends/gpu/
Dgpu_registration.cc43 tensorflow::PlatformDeviceId platform_device_id; in RegisterGpuOpHandler() local
/external/tensorflow/tensorflow/compiler/jit/
Dxla_platform_info.cc34 int32_t platform_device_id; in ParseVisibleDeviceList() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph.cc113 PlatformDeviceId platform_device_id; in GetFirstValidDeviceId() local
272 PlatformDeviceId platform_device_id; in GetEngineInfo() local
650 PlatformDeviceId platform_device_id; in GetDeviceAndAllocator() local
/external/tensorflow/tensorflow/core/grappler/costs/
Dutils.cc252 PlatformDeviceId platform_device_id; in GetDeviceInfo() local
/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/
Dtrt_engine_op.cc1306 const int platform_device_id = in AllocateCalibrationResources() local
/external/libbpf/.github/actions/build-selftests/
Dvmlinux.h21868 struct platform_device_id { struct
21869 char name[20];
21870 kernel_ulong_t driver_data;
/external/bcc/libbpf-tools/powerpc/
Dvmlinux_510.h17387 struct platform_device_id { struct
17388 char name[20];
17389 kernel_ulong_t driver_data;
Dvmlinux.h17387 struct platform_device_id { struct
17388 char name[20];
17389 kernel_ulong_t driver_data;
/external/bcc/libbpf-tools/arm64/
Dvmlinux.h15934 struct platform_device_id { struct
15935 char name[20];
15936 kernel_ulong_t driver_data;

12