def root2

bend-ml-tensor-gpu@0.1.5.1 · main.bend:447

import bend-ml-tensor-gpu@0.1.5.1/main.bend as BendMlTensorGpu
import Base

447def root2(~E: Data, ~f: @+t: U32 -> @+l: U32 -> @+e: E -> Array<F32> -> Array<F32> -> Array<F32> -> Array<F32> -> O2 -> O2, +d: Nat, +e: E, +x1: Array<F32>, +x2: Array<F32>, +x3: Array<F32>, +x4: Array<F32>, oa: Array<F32>, ob: Array<F32>) -> O2:
448  tree2(~E, ~f, d, 0, U32.pow(2, d), e, x1, x2, x3, x4, oa, ob)
449
450@unsafe

Uses (7)

Used by (2)