From bdfa6ed2d232b0629079c4d0145a12117c996cfe Mon Sep 17 00:00:00 2001
From: Xiao Yu <fishx@google.com>
Date: Mon, 8 Jul 2019 18:32:28 -0700
Subject: [PATCH] Return NOT_FOUND error if we cannot find a device for a
 kernel.

PiperOrigin-RevId: 257099659
---
 tensorflow/core/common_runtime/eager/execute.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tensorflow/core/common_runtime/eager/execute.cc b/tensorflow/core/common_runtime/eager/execute.cc
index 6a0aad55f43..a206d4068e8 100644
--- a/tensorflow/core/common_runtime/eager/execute.cc
+++ b/tensorflow/core/common_runtime/eager/execute.cc
@@ -243,7 +243,7 @@ Status SelectDevice(EagerOperation* op, const NodeDef& ndef, EagerContext* ctx,
   TF_RETURN_IF_ERROR(SupportedDeviceTypesForNode(
       ctx->prioritized_device_type_list(), ndef, &supported_devs));
   if (supported_devs.empty()) {
-    return errors::Internal("Could not find valid device for node.\nNode:",
+    return errors::NotFound("Could not find valid device for node.\nNode:",
                             FormatNodeDefForError(ndef),
                             "\nAll kernels registered for op ", ndef.op(),
                             " :\n", KernelsRegisteredForOp(ndef.op()));