Skip to main content

Module broadcast

Module broadcast 

Source

Structs§

BroadcastInputPlan
Lowering plan for broadcasting one input to an output shape.

Enums§

BroadcastError
Error returned when NumPy-style broadcast planning fails.

Functions§

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.