| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_id_utils.h | 34 se::Platform* gpu_manager, PlatformGpuId platform_gpu_id) { in ExecutorForPlatformGpuId()
38 PlatformGpuId platform_gpu_id) { in ExecutorForPlatformGpuId()
43 PlatformGpuId platform_gpu_id; in ExecutorForTfGpuId() local
51 PlatformGpuId platform_gpu_id; in CheckValidTfGpuId() local
|
| D | gpu_bfc_allocator_test.cc | 53 PlatformGpuId platform_gpu_id(0); in TEST() local
86 PlatformGpuId platform_gpu_id(0); in TEST() local
148 PlatformGpuId platform_gpu_id(0); in TEST() local
187 PlatformGpuId platform_gpu_id(0); in TEST() local
197 PlatformGpuId platform_gpu_id(0); in TEST() local
206 PlatformGpuId platform_gpu_id(0); in TEST() local
218 PlatformGpuId platform_gpu_id(0); in TEST() local
238 PlatformGpuId platform_gpu_id(0); in TEST() local
304 PlatformGpuId platform_gpu_id(0); in TEST() local
320 PlatformGpuId platform_gpu_id(0); in BM_Allocation() local
[all …]
|
| D | gpu_debug_allocator_test.cc | 37 const PlatformGpuId platform_gpu_id(0); in TEST() local
134 const PlatformGpuId platform_gpu_id(0); in TEST() local
178 const PlatformGpuId platform_gpu_id(0); in TEST() local
225 const PlatformGpuId platform_gpu_id(0); in TEST() local
235 const PlatformGpuId platform_gpu_id(0); in TEST() local
|
| D | gpu_id_manager.cc | 37 Status Insert(TfGpuId tf_gpu_id, PlatformGpuId platform_gpu_id) in Insert()
87 PlatformGpuId platform_gpu_id) { in InsertTfPlatformGpuIdPair()
92 PlatformGpuId* platform_gpu_id) { in TfToPlatformGpuId()
|
| D | gpu_id_manager_test.cc | 26 PlatformGpuId platform_gpu_id; in TfToPlatformGpuId() local
|
| D | gpu_cudamalloc_allocator.cc | 31 PlatformGpuId platform_gpu_id) in GPUcudaMallocAllocator()
|
| D | gpu_debug_allocator.cc | 79 PlatformGpuId platform_gpu_id) in GPUDebugAllocator()
157 PlatformGpuId platform_gpu_id) in GPUNanResetAllocator()
|
| D | gpu_device.cc | 133 PlatformGpuId platform_gpu_id; in Reinitialize() local
398 PlatformGpuId platform_gpu_id; in Init() local
785 int32 platform_gpu_id; in ParseVisibleDeviceList() local
888 PlatformGpuId platform_gpu_id, in SingleVirtualDeviceMemoryLimit()
1031 for (PlatformGpuId platform_gpu_id : valid_platform_gpu_ids) { in CreateDevices() local
1118 const PlatformGpuId platform_gpu_id = valid_platform_gpu_ids[i]; in CreateDevices() local
1161 static string GetShortDeviceDescription(PlatformGpuId platform_gpu_id, in GetShortDeviceDescription()
1190 PlatformGpuId platform_gpu_id; in CreateGPUDevice() local
1284 PlatformGpuId platform_gpu_id; in GetDeviceLocalities() local
|
| D | gpu_process_state.cc | 101 PlatformGpuId platform_gpu_id; in GetGPUAllocator() local
|
| D | gpu_device.h | 103 PlatformGpuId platform_gpu_id; in gpu_id() local
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | utils.cc | 77 DeviceProperties GetLocalGPUInfo(PlatformGpuId platform_gpu_id) { in GetLocalGPUInfo()
160 PlatformGpuId platform_gpu_id; in GetDeviceInfo() local
|
| D | single_machine.cc | 96 PlatformGpuId platform_gpu_id; in Provision() local
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_gpu_device.cc | 44 int32 platform_gpu_id; in ParseVisibleDeviceList() local
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | utils.cc | 244 PlatformGpuId platform_gpu_id; in GetDeviceInfo() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_op.cc | 686 const int platform_gpu_id = in AllocateCalibrationResources() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 739 PlatformGpuId platform_gpu_id; in GetDeviceAndAllocator() local
|