Searched defs:auto_sharding (Results 1 – 1 of 1) sorted by relevance
926 HloSharding* auto_sharding, HloSharding* manual_sharding) { in RefineManualAutoShardingFromAuto()998 HloSharding* auto_sharding, HloSharding* manual_sharding) { in RefineManualAutoShardingFromManual()1076 HloSharding auto_sharding = annotate_op->sharding(); in InferUnspecifiedDimsFromOperand() local1089 HloSharding auto_sharding = man_conversion_op->sharding(); in InferUnspecifiedDimsFromOperand() local1129 HloSharding auto_sharding = annotate_op->sharding(); in InferUnspecifiedDimsFromOneUser() local1141 HloSharding auto_sharding = man_conversion_op->sharding(); in InferUnspecifiedDimsFromOneUser() local