def aligned_or_rows

stelliferous@0.0.2.0 · tensor_band_concat.bend:149

import stelliferous@0.0.2.0/tensor_band_concat.bend as TensorBandConcat
import Base
import stelliferous@0.0.2.0/tensor.bend as Tensors
import stelliferous@0.0.2.0/tensor_band.bend as Bands

149def aligned_or_rows(~Element: Data,+geometry: Bands.Geometry,channel: U32,workers: U32,
150  input: Bands.Bands(~Element) & Bands.Plan,output: Bands.Bands(~Element) & Bands.Plan,shape: List<&2,U32>,fill: Element,status: Tensors.Status,previous: U32) -> Bands.Bands(~Element):
151  (input,source_plan) = input
152  (output,+target_plan) = output
153  selected_rows(~Element,Bands.same_plan(source_plan,target_plan),geometry,channel,workers,target_plan,input,output,shape,fill,status,previous)

Uses (10)

Used by (1)