Searched refs:GpuPlatformName (Results 1 – 5 of 5) sorted by relevance
32 return se::MultiPlatformManager::PlatformWithName(GpuPlatformName()).status(); in ValidateGPUMachineManager()36 auto result = se::MultiPlatformManager::PlatformWithName(GpuPlatformName()); in GPUMachineManager()38 LOG(FATAL) << "Could not find Platform with name " << GpuPlatformName(); in GPUMachineManager()45 string GpuPlatformName() { in GpuPlatformName() function
29 se::MultiPlatformManager::PlatformWithName(GpuPlatformName()) in TEST()49 se::MultiPlatformManager::PlatformWithName(GpuPlatformName()) in TEST()84 se::MultiPlatformManager::PlatformWithName(GpuPlatformName()) in TEST()147 se::MultiPlatformManager::PlatformWithName(GpuPlatformName()) in TEST()251 se::MultiPlatformManager::PlatformWithName(GpuPlatformName()) in TEST()
41 std::string GpuPlatformName();
74 se::MultiPlatformManager::PlatformWithName(tensorflow::GpuPlatformName()); in ListPhysicalDevices()123 se::MultiPlatformManager::PlatformWithName(tensorflow::GpuPlatformName()); in CreateDevices()
34 se::MultiPlatformManager::PlatformWithName(tensorflow::GpuPlatformName()) in MlirGpuTestBase()