shapes#

The shapes sub-namespace of onnx_optim hosts the per-operator shape-inference functions (ComputeShape*). Each function consumes a ShapesContext (a name → OptimTensor map), a NodeProto and the names of the input values to read, and writes the descriptors of the node’s outputs back into the context.

Concrete functions are organised per operator domain (currently only math).