experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[tfdbg2] Exclude RemoteCall from instrumentation

Browse Source 
PiperOrigin-RevId: 306736646
Change-Id: I89e1442d3e31cfa790eb293fb1a881ddaa101b7a
This commit is contained in:
Shanqing Cai 2020-04-15 15:53:22 -07:00 committed by TensorFlower Gardener
parent 612f517ea7
commit 1176a04c15
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tensorflow/python/debug/lib/op_callbacks_common.py
View File

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