controlflow/utils.h

Control-flow schema helpers for shape inference, including onnx_light::IfInferenceFunction(), onnx_light::LoopInferenceFunction(), and onnx_light::ScanInferenceFunction().

Declares control-flow shape-inference helpers shared by If, Loop, and Scan.

This header exposes utility routines used by control-flow operator schemas to validate axes, clear inferred shapes, and propagate shape/type information.

namespace onnx_light

Alias that makes onnx-light headers compatible with code that references ONNX_LIGHT_NAMESPACE (the macro used in the standard onnx package).

Set to ONNX_LIGHT_NAMESPACE so both names resolve to the same namespace.

Symbol-visibility attribute for the public onnx-light C++ API.

Defined as empty because onnx-light does not require explicit __declspec(dllexport) or __attribute__((visibility("default"))) annotations — visibility is controlled at the shared-library level. The macro is provided so that vendored ONNX headers that decorate their declarations with ONNX_API compile without modification.

Namespace alias so that ONNX C++ code (and consumers such as onnxruntime) that refers to the literal onnx namespace — rather than the ONNX_NAMESPACE macro — resolves to the onnx-light namespace. The standard onnx package lives in namespace onnx; onnx-light uses onnx_light (via ONNX_LIGHT_NAMESPACE), so this alias keeps onnx-light a true drop-in. It is only introduced when the onnx-light namespace differs from onnx.

Functions

void ClearShape(TypeProto &input_type)

int handle_negative_axis_validate(const std::string &attrib, int axis, int rank)

void IfInferenceFunction(InferenceContext &ctx)

void LoopInferenceFunction(InferenceContext &ctx)

void ScanInferenceFunction(InferenceContext &ctx)