Searched refs:GetDeviceClassForNonChannelDevice (Results 1 – 3 of 3) sorted by relevance
332 TEST(DeviceClassTest, GetDeviceClassForNonChannelDevice) { in TEST() argument334 GetDeviceClassForNonChannelDevice("SOMETHING_WEIRD_DEVICE_NAME")); in TEST()335 EXPECT_EQ("/worker/GPU", GetDeviceClassForNonChannelDevice( in TEST()337 EXPECT_EQ("/worker/CPU", GetDeviceClassForNonChannelDevice( in TEST()339 EXPECT_EQ("/worker_train/CPU", GetDeviceClassForNonChannelDevice( in TEST()341 EXPECT_EQ("//GPU", GetDeviceClassForNonChannelDevice("/device:GPU:7")); in TEST()
400 string GetDeviceClassForNonChannelDevice(const string& device_name) { in GetDeviceClassForNonChannelDevice() function435 "Channel", ": ", GetDeviceClassForNonChannelDevice(src_device_full), in GetDeviceClass()436 " -> ", GetDeviceClassForNonChannelDevice(dst_device_full)); in GetDeviceClass()438 return GetDeviceClassForNonChannelDevice(device_name); in GetDeviceClass()
116 string GetDeviceClassForNonChannelDevice(const string& device_name);