Searched refs:GPUDeviceName (Results 1 – 2 of 2) sorted by relevance
65 string GPUDeviceName(TF_Session* session) { in GPUDeviceName() function92 string GPUDeviceName() { in GPUDeviceName() function103 const string gpu_device_name = GPUDeviceName(sess); in GPUDeviceName()189 if (!GPUDeviceName().empty()) return; in TEST()1046 const string gpu_device = GPUDeviceName(); in TEST()1054 const string gpu_device = GPUDeviceName(); in TEST()2384 if (!GPUDeviceName().empty()) return; in TEST()2412 if (!GPUDeviceName().empty()) return; in TEST()
1914 string GPUDeviceName(Session* session) { in GPUDeviceName() function1927 const string gpu_device_name = GPUDeviceName(session.get()); in TEST()1999 const string gpu_device_name = GPUDeviceName(session.get()); in TestFeedAndFetchTensorsInDeviceMemory()2055 const string gpu_device_name = GPUDeviceName(session.get()); in TestFeedAndFetchTensorsInDeviceMemoryFailsToMakeCallable()