def pk_d

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

import Base

148# one block below 16384 cells (MNIST's 100 x 784 and 128 x 784 pack faster in one), else 2^d
149def pk_d(big: Bool, d: Nat) -> Nat:
150  match big:
151    case True{}:
152      d
153    case False{}:
154      0n
155
156@unsafe