Searched refs:GetDeviceClassForNonChannelDevice (Results 1 – 3 of 3) sorted by relevance
328 TEST(DeviceClassTest, GetDeviceClassForNonChannelDevice) { in TEST() argument330 GetDeviceClassForNonChannelDevice("SOMETHING_WEIRD_DEVICE_NAME")); in TEST()331 EXPECT_EQ("/worker/GPU", GetDeviceClassForNonChannelDevice( in TEST()333 EXPECT_EQ("/worker/CPU", GetDeviceClassForNonChannelDevice( in TEST()335 EXPECT_EQ("/worker_train/CPU", GetDeviceClassForNonChannelDevice( in TEST()337 EXPECT_EQ("//GPU", GetDeviceClassForNonChannelDevice("/device:GPU:7")); in TEST()
409 string GetDeviceClassForNonChannelDevice(const string& device_name) { in GetDeviceClassForNonChannelDevice() function444 "Channel", ": ", GetDeviceClassForNonChannelDevice(src_device_full), in GetDeviceClass()445 " -> ", GetDeviceClassForNonChannelDevice(dst_device_full)); in GetDeviceClass()447 return GetDeviceClassForNonChannelDevice(device_name); in GetDeviceClass()
115 string GetDeviceClassForNonChannelDevice(const string& device_name);