~/jendhub

def pk_par

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

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

157def pk_par(d: Nat, +cnt: Nat, +tk: Nat, +k: Nat, +lstep: U32, +st: U32, a: Array<F32>) -> TA.PK:
158  pk_of(par2(~PE, ~F32, ~TA.T8, ~pk_leaf, PE{tk, U32.from_nat(tk), U32.from_nat(k), lstep, st}, pk_d(Nat.is_le(16384n, Nat.mul(cnt, tk)), d), cnt, a, Array.new(TA.T8, depth(Nat.mul(cnt, tk)), TA.t8_zero())))

Uses (17)

Used by (1)