| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | utils_test.cc | 36 properties = GetLocalGPUInfo(PlatformGpuId(100)); in TEST()
40 properties = GetLocalGPUInfo(PlatformGpuId(0)); in TEST()
48 properties = GetLocalGPUInfo(PlatformGpuId(100)); in TEST()
52 properties = GetLocalGPUInfo(PlatformGpuId(0)); in TEST()
59 properties = GetLocalGPUInfo(PlatformGpuId(0)); in TEST()
62 properties = GetLocalGPUInfo(PlatformGpuId(100)); in TEST()
101 GpuIdManager::InsertTfPlatformGpuIdPair(TfGpuId(0), PlatformGpuId(100))); in TEST()
107 GpuIdManager::InsertTfPlatformGpuIdPair(TfGpuId(1), PlatformGpuId(0))); in TEST()
|
| D | utils.cc | 77 DeviceProperties GetLocalGPUInfo(PlatformGpuId platform_gpu_id) { in GetLocalGPUInfo()
160 PlatformGpuId platform_gpu_id; in GetDeviceInfo()
168 return GetLocalGPUInfo(PlatformGpuId(0)); in GetDeviceInfo()
|
| D | utils.h | 31 DeviceProperties GetLocalGPUInfo(PlatformGpuId platform_gpu_id);
|
| D | single_machine.cc | 96 PlatformGpuId platform_gpu_id; in Provision()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 135 PlatformGpuId platform_gpu_id; in Reinitialize()
476 PlatformGpuId platform_gpu_id; in Init()
781 std::vector<PlatformGpuId>* visible_gpu_order) { in ParseVisibleDeviceList()
811 visible_gpu_order->push_back(PlatformGpuId(platform_gpu_id)); in ParseVisibleDeviceList()
816 std::set<PlatformGpuId> visible_device_set(visible_gpu_order->begin(), in ParseVisibleDeviceList()
828 const std::vector<PlatformGpuId>& visible_gpu_order, in VerifyVirtualDeviceSettings()
829 const std::vector<PlatformGpuId>& valid_platform_gpu_ids, in VerifyVirtualDeviceSettings()
956 PlatformGpuId platform_gpu_id, in SingleVirtualDeviceMemoryLimit()
1096 std::vector<PlatformGpuId> visible_gpu_order(device_count); in CacheDeviceIds()
1104 for (PlatformGpuId platform_gpu_id : cached_device_ids_) { in ListPhysicalDevices()
[all …]
|
| D | gpu_virtual_mem_allocator.h | 47 stream_executor::gpu::GpuContext& gpu_context, PlatformGpuId gpu_id,
49 const std::vector<PlatformGpuId>& peer_gpu_ids);
77 stream_executor::gpu::GpuContext& gpu_context, PlatformGpuId gpu_id,
82 PlatformGpuId gpu_id_;
|
| D | gpu_debug_allocator_test.cc | 48 const PlatformGpuId platform_gpu_id(0); in TEST()
75 const PlatformGpuId platform_gpu_id(0); in TEST()
111 const PlatformGpuId platform_gpu_id(0); in TEST()
144 const PlatformGpuId platform_gpu_id(0); in TEST()
186 const PlatformGpuId platform_gpu_id(0); in TEST()
231 const PlatformGpuId platform_gpu_id(0); in TEST()
241 const PlatformGpuId platform_gpu_id(0); in TEST()
|
| D | gpu_device.h | 90 PlatformGpuId platform_gpu_id; in gpu_id()
328 std::set<std::pair<PlatformGpuId, PlatformGpuId>> directed_links;
335 const std::vector<PlatformGpuId>& visible_gpu_order,
366 Status EnablePeerAccess(const std::vector<PlatformGpuId>& visible_gpu_order);
373 Status GetValidDeviceIds(const std::vector<PlatformGpuId>& visible_gpu_order,
374 std::vector<PlatformGpuId>* ids);
389 std::vector<PlatformGpuId> cached_device_ids_;
|
| D | gpu_virtual_mem_allocator.cc | 47 PlatformGpuId gpu_id) { in CheckVirtualAddressManagementSupport()
64 GpuContext& gpu_context, PlatformGpuId gpu_id, in Create()
66 const std::vector<PlatformGpuId>& peer_gpu_ids) { in Create()
114 PlatformGpuId gpu_id, const std::vector<GpuDeviceHandle> access_gpu_handles, in GpuVirtualMemAllocator()
|
| D | gpu_debug_allocator.h | 37 PlatformGpuId platform_gpu_id);
67 PlatformGpuId platform_gpu_id);
|
| D | gpu_process_state.cc | 84 const GPUOptions& options, PlatformGpuId platform_gpu_id, in CreateSubAllocator()
105 absl::flat_hash_set<PlatformGpuId> platform_peer_gpu_ids; in CreateSubAllocator()
108 PlatformGpuId platform_gpu_id; in CreateSubAllocator()
112 std::vector<PlatformGpuId> platform_peer_gpu_ids_vec( in CreateSubAllocator()
159 PlatformGpuId platform_gpu_id; in GetGPUAllocator()
|
| D | gpu_id.h | 25 typedef PlatformDeviceId PlatformGpuId; typedef
|
| D | gpu_virtual_mem_allocator_test.cc | 38 PlatformGpuId gpu_id(0); in CreateAllocator()
51 PlatformGpuId gpu_id(0); in TEST()
|
| D | gpu_cudamalloc_allocator.h | 35 PlatformGpuId platform_gpu_id);
|
| D | gpu_debug_allocator.cc | 79 PlatformGpuId platform_gpu_id) in GPUDebugAllocator()
158 PlatformGpuId platform_gpu_id) in GPUNanResetAllocator()
|
| D | gpu_cudamalloc_allocator.cc | 30 PlatformGpuId platform_gpu_id) in GPUcudaMallocAllocator()
|
| D | gpu_device_test.cc | 33 int64 GetTotalGPUMemory(PlatformGpuId gpu_id) { in GetTotalGPUMemory()
43 Status GetComputeCapability(PlatformGpuId gpu_id, int* cc_major, in GetComputeCapability()
353 TF_ASSERT_OK(GetComputeCapability(PlatformGpuId(0), &cc_major, &cc_minor)); in TEST_F()
374 static constexpr PlatformGpuId kPlatformGpuId(0); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph.cc | 105 std::pair<TfGpuId, PlatformGpuId> GetFirstValidDeviceId() { in GetFirstValidDeviceId()
108 PlatformGpuId platform_gpu_id; in GetFirstValidDeviceId()
117 return std::make_pair(TfGpuId(-1), PlatformGpuId(-1)); in GetFirstValidDeviceId()
270 PlatformGpuId platform_gpu_id; in GetEngineInfo()
639 PlatformGpuId platform_gpu_id; in GetDeviceAndAllocator()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | utils.cc | 245 PlatformGpuId platform_gpu_id; in GetDeviceInfo()
249 platform_gpu_id = PlatformGpuId(parsed.id); in GetDeviceInfo()
|