Consider beyond immediate neighbors to find exit node.
authorJacques Pienaar <jpienaar@google.com>
Fri, 2 Feb 2018 18:04:15 +0000 (10:04 -0800)
committerTensorFlower Gardener <gardener@tensorflow.org>
Fri, 2 Feb 2018 19:03:51 +0000 (11:03 -0800)
commitb70a3794b6e06418b6d5b4d7142edeb78494fe7b
treec7d95bf7fff9c04a5a246ccd1ef4f0da790530c5
parentb38575ea9b12534e9c63833782a46e43e1c9a8af
Consider beyond immediate neighbors to find exit node.

Most of the exit nodes are immediate neighbors of the switch, except we do have
cases where the switch feeds into an identity that feeds into a exit.

PiperOrigin-RevId: 184297180
tensorflow/compiler/tf2xla/functionalize_control_flow.cc