Commit 9018715a authored by Priya Gupta's avatar Priya Gupta Committed by TensorFlower Gardener
Browse files

MNIST eager dist strat example: Add an option to run the step in a... 

MNIST eager dist strat example: Add an option to run the step in a tf.function. Also changed the example to run with TF 2.0 from compat.

PiperOrigin-RevId: 232073692
parent 9167d6d1
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