axis_input_to_attribute.h#

namespace ONNX_LIGHT_NAMESPACE
namespace version_conversion
class AxisInputToAttribute : public ONNX_LIGHT_NAMESPACE::version_conversion::Adapter#
#include <axis_input_to_attribute.h>

Public Functions

inline explicit AxisInputToAttribute(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 bool HasAxisInput(const Node *node) const#
inline void HandleConstantNode(Node *node, Node *axis_node, Value *axis_val) const#
inline void HandleInitializerNode(const std::shared_ptr<Graph> &graph, Node *node, Value *axis_val) const#
inline Node *EnsureAndReturnNode(Node *node) const#

Private Members

size_t axis_index#
int64_t default_axis#