Home
last modified time | relevance | path

Searched refs:ValidateGPUMachineManager (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_init.h30 Status ValidateGPUMachineManager();
H A Dgpu_init.cc31 Status ValidateGPUMachineManager() { in ValidateGPUMachineManager() function
H A Dgpu_device.cc1212 TF_RETURN_IF_ERROR(ValidateGPUMachineManager()); in CacheDeviceIds()
1249 TF_RETURN_IF_ERROR(ValidateGPUMachineManager()); in GetDeviceDetails()
1271 TF_RETURN_IF_ERROR(ValidateGPUMachineManager()); in CreateDevices()
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/
H A Ddevices.cc43 if (ValidateGPUMachineManager().ok()) { in GetNumAvailableGPUs()