--- /dev/null
+#ifndef __NEURUN_GRAPH_OPERATION_SOFTMAX_H__
+#define __NEURUN_GRAPH_OPERATION_SOFTMAX_H__
+
+#include <memory>
+
+#include "graph/operation/Node.h"
+#include "internal/op/Softmax.h"
+
+namespace neurun
+{
+namespace graph
+{
+namespace operation
+{
+namespace Softmax
+{
+
+class Node : public graph::operation::Node
+{
+public:
+ Node(std::unique_ptr<::internal::tflite::op::Softmax::Node> &&op) : _op{std::move(op)} {}
+
+public:
+ virtual operand::IndexSet inputs() const override { return {_op->param().input_index}; }
+ virtual operand::IndexSet outputs() const override { return {_op->param().output_index}; }
+ virtual const ::internal::tflite::op::Node *op() const override { return _op.get(); }
+
+private:
+ std::unique_ptr<::internal::tflite::op::Softmax::Node> _op;
+};
+
+} // namespace Softmax
+} // namespace operation
+} // namespace graph
+} // namespace neurun
+
+#endif // __NEURUN_GRAPH_OPERATION_SOFTMAX_H__