def depth2
import bend-ml-tensor-gpu@0.1.5.1/main.bend as BendMlTensorGpuimport Base
import bend-ml-tensor-array@0.1.19.0/main.bend as TA
351def depth2(big: Bool, +n: Nat) -> Nat:
352 match big:
353 case True{}: 14n
354 case False{}: TA.cap_depth(n)