Make init_scope preserve the inner device stack when lifting into a graph.
Eager execution doesn't implement device stacks and in particular it doesn't support device functions (which determine the device on a per-op basis), so in general it's not possible to do the same when lifting into the eager context. PiperOrigin-RevId: 197583446
Loading
Please sign in to comment