axis_attribute_to_input.h#

namespace ONNX_LIGHT_NAMESPACE
namespace version_conversion
class AxisAttributeToInput : public ONNX_LIGHT_NAMESPACE::version_conversion::Adapter#
#include <axis_attribute_to_input.h>

Public Functions

inline AxisAttributeToInput(const std::string &op_name, const OpSetID &initial, const OpSetID &target, size_t axis_index, int64_t default_axis)#
inline virtual Node *adapt(std::shared_ptr<Graph> graph, Node *node) const override#

Private Functions

inline void AttrToInput(const std::shared_ptr<Graph> &graph, Node *node, int64_t axis, size_t axis_index) const#
inline Node *CreateAxisInput(const std::shared_ptr<Graph> &graph, Node *node, int64_t axis) const#

Private Members

size_t axis_index#
int64_t default_axis#