def St.plan_shape

bendtorch-load@1.3.0.0 · main.bend:465

import Base

465def St.plan_shape(ok: Bool, +name: String, f32: Bool, +start: W64, +sh: List<&2, W64>, +want: List<&2, U32>, of: List<&2, W64>) -> Plan:
466  match ok f32:
467    case True{} True{}: St.plan_of(name, False{}, start, St.prod(want), 4, of)
468    case True{} False{}: St.plan_of(name, True{}, start, St.prod(want), 2, of)
469    case False{} _: PlanBad{"tensor " ++ name ++ ": shape [" ++ St.show_w(sh, "") ++ "] in the file, [" ++ St.show_u(want, "") ++ "] expected"}