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

Support native multi-device functions 

Implementation mostly follows the code of PartitionCallOp.

This change also fixes a bug in PartitionCallOp where the
instantiated functions were never released.

PiperOrigin-RevId: 227088326
parent add19e0e
Expand all 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