From 1176a04c15e4509eea0978bcc2fb0e2d01088f07 Mon Sep 17 00:00:00 2001
From: Shanqing Cai <cais@google.com>
Date: Wed, 15 Apr 2020 15:53:22 -0700
Subject: [PATCH] [tfdbg2] Exclude RemoteCall from instrumentation

PiperOrigin-RevId: 306736646
Change-Id: I89e1442d3e31cfa790eb293fb1a881ddaa101b7a
---
 tensorflow/python/debug/lib/op_callbacks_common.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tensorflow/python/debug/lib/op_callbacks_common.py b/tensorflow/python/debug/lib/op_callbacks_common.py
index 1848bd43a3a..279411721a1 100644
--- a/tensorflow/python/debug/lib/op_callbacks_common.py
+++ b/tensorflow/python/debug/lib/op_callbacks_common.py
@@ -35,6 +35,10 @@ OP_CALLBACK_SKIP_OPS = (
     b"StatefulPartitionedCall",
     b"Switch",
     b"While",
+    # NOTE(b/154097452): On TPUs, debugger ops are colocated with RemoteCall
+    # ops. This exclusion prevents an error due to no OpKernel for those
+    # debugger ops.
+    b"RemoteCall",
     # TPU-specific ops begin.
     b"TPUReplicatedInput",
     b"TPUReplicateMetadata",