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 function

replicated_rows

replicated_rows()

max.experimental.sharding.replicated_rows(*args, output_count=1)

source

Returns the one row of an op that runs only on whole tensors.

Parameters:

• *args (Any) – The op’s arguments, as the rule receives them.
• output_count (int) – The number of results. Defaults to 1.

Returns:

A single row placing every operand and result Replicated.

Raises:

ValueError – If args holds no tensor operand.

Return type:

list[AxisAssignment]