Searched refs:expected_device (Results 1 – 3 of 3) sorted by relevance
147 const Device* expected_device, in ExtractTensorFromEagerTensor() argument155 if (expected_device == actual_device) return Status::OK(); in ExtractTensorFromEagerTensor()156 const string& expected_device_name = expected_device->attributes().name(); in ExtractTensorFromEagerTensor()158 if (!IsCPUDevice(expected_device)) { in ExtractTensorFromEagerTensor()
179 Device* expected_device = kernel->InputDevice(i); in ValidateInputTypeAndPlacement() local183 if (expected_device != handle_device && maybe_copy) { in ValidateInputTypeAndPlacement()186 expected_device, &handle)); in ValidateInputTypeAndPlacement()
513 expected_device = "/device:CPU:100"515 self.assertEqual(expected_device, v1.op.device)