Searched defs:assign_tuple_sharding (Results 1 – 2 of 2) sorted by relevance
166 assign_tuple_sharding=False, argument261 def replicate(tensor, assign_tuple_sharding=False, use_sharding_op=False): argument270 assign_tuple_sharding=False, argument281 assign_tuple_sharding=False, argument301 assign_tuple_sharding=False, argument
1072 xla::XlaScopedShardingAssignment assign_tuple_sharding( in BuildArguments() local