Searched refs:expected_device (Results 1 – 3 of 3) sorted by relevance
177 const Device* expected_device, in ExtractTensorFromEagerTensor() argument183 if (expected_device == actual_device) return Status::OK(); in ExtractTensorFromEagerTensor()184 const string& expected_device_name = expected_device->attributes().name(); in ExtractTensorFromEagerTensor()186 if (!IsCPUDevice(expected_device)) { in ExtractTensorFromEagerTensor()
190 const Device* expected_device = kernel->InputDevice(i); in ValidateInputTypeAndPlacement() local192 op, op_device_name, i, expected_device, run_metadata, in ValidateInputTypeAndPlacement()
484 expected_device = "/device:CPU:100"486 self.assertEqual(expected_device, v1.op.device)