def size

stelliferous@0.0.2.0 · tensor_band_schedule.bend:19

import stelliferous@0.0.2.0/tensor_band_schedule.bend as TensorBandSchedule

import Base
import stelliferous@0.0.2.0/tensor_band.bend as Bands

19def size(layout: Bands.Plan) -> Size:
20  match layout:
21    case Bands.Rows{first,+rows}: Size{1,rows,rows}
22    case Bands.Split{left,right}: merge(size(left),size(right))

Uses (3)

Used by (1)

stelliferous@0.0.2.0/tensor_band_schedule:parallel