tree 13e101adc28441389b15f978642af899f2814e1f
parent ce7fef0740ebe0e7f92ba1709e1ce04cb5f59b2c
author Andy Ly <lyandy@google.com> 1592619495 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1592619760 -0700

Remove assertions checking device map size in TPUVariableRuntimeReformattingPass.

It is possible for more than one replicated device to be populated (e.g. replicated host).

PiperOrigin-RevId: 317419924
Change-Id: I1d8fa95654f324557d78adf4570adc6c3cfdabb4