Commit 48eb1be4 authored by Igor Ganichev's avatar Igor Ganichev Committed by TensorFlower Gardener
Browse files

Fix output device inference in ProcessFunctionLibraryRuntime 

Output device inference code only looked at requested devices as it
traced up from the _Retval node. This change makes it also look at
assigned devices.

Also, add some debugging log statements.

PiperOrigin-RevId: 238256609
parent be7d219a
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment