| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| H A D | gpu_debug_allocator_test.cc | 41 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
|
| H A D | gpu_device.cc | 129 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 …]
|
| H A 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()
|
| H A D | gpu_process_state.cc | 99 const GPUOptions& options, PlatformDeviceId platform_device_id, in CreateSubAllocator()
174 PlatformDeviceId platform_device_id; in GetGPUAllocator() local
|
| H A D | gpu_debug_allocator.cc | 79 PlatformDeviceId platform_device_id) in GPUDebugAllocator()
158 PlatformDeviceId platform_device_id) in GPUNanResetAllocator()
|
| H A D | gpu_cudamalloc_allocator.cc | 30 PlatformDeviceId platform_device_id) { in GPUcudaMallocAllocator()
|
| H A D | gpu_cudamallocasync_allocator.cc | 101 PlatformDeviceId platform_device_id, size_t pool_size, bool reserve_memory, in GpuCudaMallocAsyncAllocator()
|
| H A D | gpu_device.h | 114 PlatformDeviceId platform_device_id; in gpu_id() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/device/ |
| H A D | device_id_utils.h | 37 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
|
| H A D | device_id_manager.cc | 38 PlatformDeviceId platform_device_id) TF_LOCKS_EXCLUDED(mu_) { in Insert()
99 PlatformDeviceId platform_device_id) { in InsertTfPlatformDeviceIdPair()
106 PlatformDeviceId* platform_device_id) { in TfToPlatformDeviceId()
|
| H A D | device_id_manager_test.cc | 27 PlatformDeviceId platform_device_id; in TfToPlatformDeviceId() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/ |
| H A D | pluggable_device_factory.cc | 80 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
|
| H A D | pluggable_device_process_state.cc | 101 PlatformDeviceId platform_device_id; in GetPluggableDeviceAllocator() local
|
| H A D | pluggable_device.cc | 187 PlatformDeviceId platform_device_id; in Init() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/clusters/ |
| H A D | utils.cc | 77 DeviceProperties GetLocalGPUInfo(PlatformDeviceId platform_device_id) { in GetLocalGPUInfo()
160 PlatformDeviceId platform_device_id; in GetDeviceInfo() local
|
| H A D | single_machine.cc | 96 PlatformDeviceId platform_device_id; in Provision() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_graph.cc | 113 PlatformDeviceId platform_device_id; in GetFirstValidDeviceId() local
272 PlatformDeviceId platform_device_id; in GetEngineInfo() local
650 PlatformDeviceId platform_device_id; in GetDeviceAndAllocator() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/backends/gpu/ |
| H A D | gpu_registration.cc | 43 tensorflow::PlatformDeviceId platform_device_id; in RegisterGpuOpHandler() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | xla_platform_info.cc | 34 int32_t platform_device_id; in ParseVisibleDeviceList() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/ |
| H A D | utils.cc | 252 PlatformDeviceId platform_device_id; in GetDeviceInfo() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| H A D | trt_engine_op.cc | 1306 const int platform_device_id = in AllocateCalibrationResources() local
|
| /aosp_15_r20/external/libbpf/.github/actions/build-selftests/ |
| H A D | vmlinux.h | 38109 struct platform_device_id { struct
38132 const struct platform_device_id *id_entry; argument
|
| /aosp_15_r20/external/bcc/libbpf-tools/powerpc/ |
| H A D | vmlinux.h | 9967 struct platform_device_id { struct
9983 const struct platform_device_id *id_entry; argument
|
| H A D | vmlinux_600.h | 9967 struct platform_device_id { struct
9983 const struct platform_device_id *id_entry; argument
|
| /aosp_15_r20/external/bcc/libbpf-tools/loongarch/ |
| H A D | vmlinux.h | 35686 struct platform_device_id { struct
35687 char name[20];
35688 kernel_ulong_t driver_data;
|