Namespace tensorflow.python.tpu.tpu_sharding Classes ShardingPolicy tpu_sharding Interfaces IShardingPolicy