IMPORTANT: To view this page as Markdown, append `.md` to the URL (e.g. /get-started.md). For the complete documentation index, see llms.txt.
Skip to main content
For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).

Python class

ReduceOp

ReduceOp

class max.experimental.sharding.ReduceOp(value, names=<not given>, *values, module=None, qualname=None, type=None, start=1, boundary=None)

source

Bases: str, Enum

Reduction operations for partial placements.

Only SUM is currently implemented in eager dispatch; the others are reserved for forward compatibility.

AVG

AVG = 'avg'

source

MAX

MAX = 'max'

source

MIN

MIN = 'min'

source

SUM

SUM = 'sum'

source