Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_device.cc1213 se::Platform* gpu_manager = GPUMachineManager(); in CacheDeviceIds() local
1250 se::Platform* gpu_manager = GPUMachineManager(); in GetDeviceDetails() local
1272 se::Platform* gpu_manager = GPUMachineManager(); in CreateDevices() local
1574 se::Platform* gpu_manager = GPUMachineManager(); in CreateGPUDevice() local
1652 se::Platform* gpu_manager, std::vector<InterconnectMap>* maps) { in GetInterconnectMaps()
1685 se::Platform* gpu_manager = GPUMachineManager(); in GetDeviceLocalities() local
1753 se::Platform* gpu_manager, in GetDefaultMinGPUMultiprocessorCount()
1778 se::Platform* gpu_manager, in GetMinGPUMultiprocessorCount()
1847 se::Platform* gpu_manager = GPUMachineManager(); in EnablePeerAccess() local
1892 se::Platform* gpu_manager = GPUMachineManager(); in GetValidDeviceIds() local
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/
H A Ddevices.cc44 se::Platform* gpu_manager = GPUMachineManager(); in GetNumAvailableGPUs() local