Commit 4f464687 authored by Igor Saprykin's avatar Igor Saprykin Committed by TensorFlower Gardener
Browse files

Make replicate_model_fn more flexible to simplify users' model_fn. 

If there is only one device then replication/aggregation overhead isn't added.  It is okay to not use TowerEstimator if there is only one device.  It is okay to use TowerEstimator but not use replicate_model_fn.

PiperOrigin-RevId: 181388296
parent 605a0b85
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