Commit 3295fdb5 authored by Priya Gupta's avatar Priya Gupta Committed by TensorFlower Gardener
Browse files

MirroredStrategy: Throw an error if there is a merge call in the middle of a...

MirroredStrategy: Throw an error if there is a merge call in the middle of a `fn` passed to call_for_each_replica which changes the graph being used while calling `fn`. This is because we switch from a replica thread to the main thread during a merge_call and changing the graph can have undesirable behavior currently.

PiperOrigin-RevId: 236022706
parent 41949996
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment