Commit 76c9358e authored by Akshay Agrawal's avatar Akshay Agrawal Committed by TensorFlower Gardener
Browse files

Minor touch ups to PartitionedCallOp. 

Mostly just cosmetic refactoring to make PartitionedCallOp more readable; also registers a GPU kernel.

PiperOrigin-RevId: 199433460
parent 95cd2d44
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