tree dbf6f183a1c148c6889ea25ea5f939c6b88a659a
parent 3cc504a901fdf592e1a8bd9a01e93d34c8d30a88
author Jeremy Meredith <jsmeredith@google.com> 1631241239 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1631241539 -0700

Adding checks for control flow nodes when calculating device colocation within the graph. In addition to the prior fix to colocation_graph.cc, this fixes the issue underlying some skipped GPU-enabled tests in placement_test.py; this commit re-enables those skipped placements tests.

PiperOrigin-RevId: 395838212
Change-Id: I47e99b471196be5167a075b5ee17e1bd5504e92c