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
Loading
Please sign in to comment