defs/controlflow/utils.h

0001 /*
0002  * SPDX-License-Identifier: Apache-2.0
0003  */
0004
0005 #pragma once
0006
0007 #include <string>
0008
0009 #include "onnx/defs/schema.h"
0010
0011 namespace ONNX_NAMESPACE {
0012
0013 void ClearShape(TypeProto& input_type);
0014
0015 int handle_negative_axis_validate(const std::string& attrib, int axis, int rank);
0016
0017 void IfInferenceFunction(InferenceContext& ctx);
0018
0019 void LoopInferenceFunction(InferenceContext& ctx);
0020
0021 void ScanInferenceFunction(InferenceContext& ctx);
0022
0023 } // namespace ONNX_NAMESPACE