def capacity_of

bendtorch-core@0.1.0.0 · src/autograd/reduce_PROOF.bend:166

import bendtorch-core@0.1.0.0/src/autograd/reduce_PROOF.bend as ReducePROOF
import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape

166def capacity_of(+shape: Shape.Shape) -> Shape.LE(Shape.logical_numel(Shape.dimensions(shape)), 1048576n):
167  match shape:
168    case Shape.SShape{dims, size, exact, rank_bound, capacity_bound}:
169      capacity_bound

Uses (4)

Used by (2)