Module broadcast 

Structs

BroadcastInputPlan

Lowering plan for broadcasting one input to an output shape.

Enums

BroadcastError

Error returned when NumPy-style broadcast planning fails.

Functions

broadcast_error_to_validation

Convert a broadcast-planning failure to the shared validation vocabulary.

broadcast_in_dim_extent_error

Find a broadcast extent failure for an already structurally validated BroadcastInDim mapping.

broadcast_input_plan

Plan how one input should lower to BroadcastInDim.

broadcast_shape

Compute the NumPy-style broadcast shape for two concrete shapes.

broadcast_shapes

Compute the common NumPy-style broadcast shape for zero or more shapes.