Searched defs:assign_tuple_sharding (Results 1 – 2 of 2) sorted by relevance
207 assign_tuple_sharding=False, argument308 def replicate(tensor, assign_tuple_sharding=False, use_sharding_op=False): argument317 assign_tuple_sharding=False, argument328 assign_tuple_sharding=False, argument351 assign_tuple_sharding=False, argument
1099 xla::XlaScopedShardingAssignment assign_tuple_sharding( in BuildArguments() local