~/jendhub

def t8_tree

bend-ml-tensor-fork@0.1.6.0 · main.bend:199

import bend-ml-tensor-fork@0.1.6.0/main.bend as BendMlTensorFork
import Base
import bend-ml-tensor-array@0.1.19.0/main.bend as TA

199def t8_tree(byrows: Bool, +ac: Bool, d: Nat, +n: Nat, +m: Nat, +tk: Nat, +tku: U32, +rot: Nat, pa: Array<TA.T8> & Array<TA.T8>, pb: Array<TA.T8> & Array<TA.T8>, pc: Array<F32> & Array<F32>) -> Array<F32>:
200  match byrows:
201    case True{}:
202      rpar(ac, d, n, m, tk, tku, rot, 0, U32.from_nat(m), pa, pb, pc)
203    case False{}:
204      cpar(ac, d, m, n, tk, tku, rot, 0, U32.from_nat(m), pa, pb, pc)

Uses (10)

Used by (1)