Commit f481d2be authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

Make embedding lookup to always return a colocation constraint free result. 

Currently, whenever embedding variable has < 2 partitions the returned result
may have a constraint on embedding var. This in turn may force the ops that use the embedding lookup result onto ps tasks.

PiperOrigin-RevId: 210456452
parent f577ae97
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