Searched defs:manual_sharding (Results 1 – 3 of 3) sorted by relevance
449 manual_sharding, argument477 manual_sharding, argument
926 HloSharding* auto_sharding, HloSharding* manual_sharding) { in RefineManualAutoShardingFromAuto()998 HloSharding* auto_sharding, HloSharding* manual_sharding) { in RefineManualAutoShardingFromManual()1077 HloSharding manual_sharding = man_conversion_op->sharding(); in InferUnspecifiedDimsFromOperand() local1088 HloSharding manual_sharding = annotate_op->sharding(); in InferUnspecifiedDimsFromOperand() local1130 HloSharding manual_sharding = man_conversion_op->sharding(); in InferUnspecifiedDimsFromOneUser() local1140 HloSharding manual_sharding = annotate_op->sharding(); in InferUnspecifiedDimsFromOneUser() local
5062 const OpSharding& manual_sharding, in ConvertSpmdFullToShardShape()5107 const OpSharding& manual_sharding, in ConvertSpmdShardToFullShape()