shape_check.h#

Shared precondition helpers reused by every per-operator ComputeShape* shape-inference function.

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.

namespace onnx_optim
namespace shapes

Functions

void CheckNodeOpAndOutput(const NodeProto &node, const char *expected_op_type, const char *caller)#

Validates the common preamble of every ComputeShape* function:

  • node.op_type() must equal expected_op_type;

  • node must declare at least one output.

The caller argument is embedded verbatim in the exception messages so that callers can surface their own name (typically the ComputeShape* function performing the check).

Throws:

std::invalid_argument – when node.op_type() differs from expected_op_type or when node.output_size() < 1.