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

Support function calls through PartitionedCall in tf2xla 

Also, make eager runtime always emit PartitionedCall and remove
special handling of xla compilation.

Becase this change makes XLA look inside PartitionedCalls, this change
had to update/disable some tests that include PartitionedCalls with
some uncompilable ops inside.

PiperOrigin-RevId: 237486703
parent 745b6aa4
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