def discover_compatible

stelliferous@0.0.2.0 · tensor_band_concat.bend:73

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

73def discover_compatible(~Element: Data,parts: Tensors.Parts<Element>) -> (Tensors.Parts<Element> & Maybe<&2,Bands.Plan>) & Bool:
74  match parts:
75    case Tensors.Single{tensor}: single_compatible(~Element,tensor_plan(~Element,tensor))
76    case Tensors.Cons{tensor,rest}: cons_compatible(~Element,tensor_plan(~Element,tensor),discover_compatible(~Element,rest))

Uses (8)

Used by (1)