Python class
Partial
Partial
class max.experimental.sharding.Partial(reduce_op=ReduceOp.SUM)
Bases: Placement
Every device holds a partial result that must be reduced.
Cached per reduce_op, so Partial() is Partial() is True.
-
Parameters:
-
reduce_op (ReduceOp) – The reduction operation to apply. Defaults to sum.
-
Return type:
Returns a cached Partial instance for the given reduce_op.
reduce_op
property reduce_op: ReduceOp
The reduction operation applied when combining partial results.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!