controlflow/utils.h

Control-flow schema helpers for shape inference, including onnx::IfInferenceFunction(), onnx::LoopInferenceFunction(), and onnx::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_NAMESPACE

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)