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",