Mojo function
shape_div
shape_div[: ImmutableOrigin, : ImmutableOrigin, //, a_t: IntTuple[$1], b_t: IntTuple[$0]](a: RuntimeTuple[a_t, element_bitwidth=element_bitwidth, unsigned=unsigned], b: RuntimeTuple[b_t, element_bitwidth=element_bitwidth, unsigned=unsigned]) -> RuntimeTuple[shape_div[::Origin[::Bool(a_t, b_t)]
Performs specialized shape division between RuntimeTuples.
This function implements a special division operation specifically designed for tensor shape calculations. Unlike standard division, it handles special cases:
- If shapes are directly divisible (a % b == 0), returns a standard division (a // b)
- If shapes are inversely divisible (b % a == 0), returns the signed reciprocal
- If shapes are incompatible, aborts with an error
This operation is essential for transformations between tensor layouts and computing broadcasting semantics.
Parameters:
- a_t (
IntTuple[$1]): Type of the first operand. - b_t (
IntTuple[$0]): Type of the second operand.
Args:
- a (
RuntimeTuple[a_t, element_bitwidth=element_bitwidth, unsigned=unsigned]): The dividendRuntimeTuple. - b (
RuntimeTuple[b_t, element_bitwidth=element_bitwidth, unsigned=unsigned]): The divisorRuntimeTuple.
Returns:
A new RuntimeTuple containing the result of the shape division.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!