def append_storage

stelliferous@0.0.2.0 · tensor_band_concat.bend:155

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

155def append_storage(~Element: Data,+geometry: Bands.Geometry,channel: U32,workers: U32,input: Tensors.Tensor<Element>,output: Bands.Bands(~Element)) -> Bands.Bands(~Element):
156  match input:
157    case Tensors.Tensor{storage,shape,fill,status,previous}:
158      appended_rows(~Element,append_tree(~Element,output,geometry,channel,Tensors.Tensor{storage,shape,fill,status,previous}))
159    case Tensors.Banded{bands,layout,shape,fill,status,previous}:
160      aligned_or_rows(~Element,geometry,channel,workers,Bands.plan(~Element,bands),Bands.plan(~Element,output),shape,fill,status,previous)