BendMlTensorFork.pk_d: def of bend-ml-tensor-fork@0.1.6.0
Import: import bend-ml-tensor-fork@0.1.6.0/main.bend as BendMlTensorFork
Lines 148-156 of bend-ml-tensor-fork@0.1.6.0/main.bend: https://jend.leeeet.dev/read.txt?path=bend-ml-tensor-fork@0.1.6.0/main.bend&start_line=148&end_line=156

import Base

# one block below 16384 cells (MNIST's 100 x 784 and 128 x 784 pack faster in one), else 2^d
def pk_d(big: Bool, d: Nat) -> Nat:
  match big:
    case True{}:
      d
    case False{}:
      0n

@unsafe

Uses (2):
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Bool
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Nat