Commit 1a1e50ee authored by Allen Lavoie's avatar Allen Lavoie Committed by TensorFlower Gardener
Browse files

init_scope uses the innermost nonempty device stack

Since we no longer inherit devices inside functions, using the inner context's device doesn't make much sense. We were also only doing it when graph building.

Should remove one more difference between Defun/(defun/function) for the tf.data migration.

PiperOrigin-RevId: 231260238
parent 37d538a6
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment