Commit 0a359212 authored by Akshay Agrawal's avatar Akshay Agrawal Committed by TensorFlower Gardener
Browse files

Fix incorrect early-out in partitioning. 

Fixes a bug in which the arg and retval node indices weren't rewritten when
a function was repartitioned by a StatefulPartitionedCallOp. Repartitioning can
occur when the kernel is visited with a previously unseen
FunctionLibraryRuntime.

PiperOrigin-RevId: 208255736
parent b4b56abe
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