Searched defs:manual_sharding (Results 1 – 4 of 4) sorted by relevance
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
449 manual_sharding, argument477 manual_sharding, argument
1270 ManualSharding manual_sharding = 2; field
5062 const OpSharding& manual_sharding, in ConvertSpmdFullToShardShape()5107 const OpSharding& manual_sharding, in ConvertSpmdShardToFullShape()