tree 458d7273cf70788cb1e59a2eac0a9e2e23649fa8
parent 952c8f8c13e36fd1d9eb0a24c49eee088d94a2d5
author A. Unique TensorFlower <gardener@tensorflow.org> 1623338170 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1623338505 -0700

Allow enable_auto_xla_input_sharding=true and enable_variable_deduplication=true for single core models.

If num_cores_per_replica==1, XLA won't split the model across multiple cores making enable_auto_xla_input_sharding effectively a no-op.

PiperOrigin-RevId: 378654009
Change-Id: Icb3bba8e099ec2a9d8081427480966fffb520967