Allow XLAShardingAdafactorOptimizer to handle tf.IndexedSlices.
Allow XLAShardingAdafactorOptimizer to handle tf.IndexedSlices.

PiperOrigin-RevId: 360594586

tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo,tensorflow/lingvo