Searched refs:ShardingPolicy (Results 1 – 3 of 3) sorted by relevance
32 p1 = tpu_sharding.ShardingPolicy()37 p2 = tpu_sharding.ShardingPolicy()46 p1 = tpu_sharding.ShardingPolicy()55 p1 = tpu_sharding.ShardingPolicy()64 p1 = tpu_sharding.ShardingPolicy()67 p2 = tpu_sharding.ShardingPolicy()71 p1 = tpu_sharding.ShardingPolicy()83 p1 = tpu_sharding.ShardingPolicy()92 p = tpu_sharding.ShardingPolicy()112 p = tpu_sharding.ShardingPolicy()[all …]
31 class ShardingPolicy(object): class
197 tpu_sharding.ShardingPolicy()