tree 7b105724a62b7cbf0cc33158405d043d2376186c
parent a4fb4cb6d4440212fbd2e694bbe4d16f02708384
author Andy Ly <lyandy@google.com> 1590693668 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1590693937 -0700

Fix bug in detecting ops in cluster when extracting head computation in TPUExtractHeadTailOutsideCompilation pass.

Check for if an op is in a cluster has been updated using cluster op as an ancestor.

PiperOrigin-RevId: 313636229
Change-Id: I200cccb038c48e1c7b0463d680fd6e3c5958deba