Searched defs:manual_sharding (Results 1 – 1 of 1) sorted by relevance
370 def auto_to_manual_spmd_partition(tensor, manual_sharding): argument389 def manual_to_auto_spmd_partition(tensor, manual_sharding, full_shape): argument