Searched refs:expected_device_name (Results 1 – 2 of 2) sorted by relevance
184 const string& expected_device_name = expected_device->attributes().name(); in ExtractTensorFromEagerTensor() local189 expected_device_name, in ExtractTensorFromEagerTensor()
1125 def test_device_name(self, device_address, expected_device_name): argument1147 'expected_device_name': expected_device_name,1149 return self.discovered_device_name == expected_device_name