TensorSelect.dispatch: def of stelliferous@0.0.2.0
Import: import stelliferous@0.0.2.0/tensor_select.bend as TensorSelect
Lines 58-66 of stelliferous@0.0.2.0/tensor_select.bend: https://jend.leeeet.dev/read.txt?path=stelliferous@0.0.2.0/tensor_select.bend&start_line=58&end_line=66

import Base
import stelliferous@0.0.2.0/tensor.bend as Tensors
import stelliferous@0.0.2.0/tensor_band_schedule.bend as Schedule

def dispatch(layout: Maybe<&2,BandPlan>,shape: List<&2,U32>,count: U32,fill: F32,workers: U32,
  mask: Tensors.Tensor<F32>,yes: Tensors.Tensor<F32>,no: Tensors.Tensor<F32>) -> Tensors.Tensor<F32>:
  match layout:
    case None{}: dense(shape,count,workers,mask,yes,no)
    case Some{BandPlan{+geometry,+layout}}:
      +workers = workers
      Tensors.Banded{selected_bands(geometry,layout,Schedule.parallel(geometry,workers,16.0,0.875,layout),
        Tensors.aligned(~F32,geometry,layout,mask),Tensors.aligned(~F32,geometry,layout,yes),Tensors.aligned(~F32,geometry,layout,no)),
        geometry,shape,fill,Tensors.Usable{},workers}

Uses (11):
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=F32
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=List
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Maybe
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=U32
  https://jend.leeeet.dev/definition.txt?path=stelliferous@0.0.2.0/tensor.bend&name=Status
  https://jend.leeeet.dev/definition.txt?path=stelliferous@0.0.2.0/tensor.bend&name=Tensor
  https://jend.leeeet.dev/definition.txt?path=stelliferous@0.0.2.0/tensor.bend&name=aligned
  https://jend.leeeet.dev/definition.txt?path=stelliferous@0.0.2.0/tensor_band_schedule.bend&name=parallel
  https://jend.leeeet.dev/definition.txt?path=stelliferous@0.0.2.0/tensor_select.bend&name=BandPlan
  https://jend.leeeet.dev/definition.txt?path=stelliferous@0.0.2.0/tensor_select.bend&name=dense
  https://jend.leeeet.dev/definition.txt?path=stelliferous@0.0.2.0/tensor_select.bend&name=selected_bands