Home
last modified time | relevance | path

Searched defs:gpu_manager (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_id_utils.h34 se::Platform* gpu_manager, PlatformGpuId platform_gpu_id) { in ExecutorForPlatformGpuId()
Dgpu_device.cc771 se::Platform* gpu_manager = GPUMachineManager(); in ParseVisibleDeviceList() local
982 se::Platform* gpu_manager = GPUMachineManager(); in CreateDevices() local
1257 se::Platform* gpu_manager, std::vector<InterconnectMap>* maps) { in GetInterconnectMaps()
1353 se::Platform* gpu_manager, in GetDefaultMinGPUMultiprocessorCount()
1379 se::Platform* gpu_manager, in GetMinGPUMultiprocessorCount()
1515 se::Platform* gpu_manager = GPUMachineManager(); in GetValidDeviceIds() local
/external/tensorflow/tensorflow/core/grappler/
Ddevices.cc34 se::Platform* gpu_manager = GPUMachineManager(); in GetNumAvailableGPUs() local