#ifndef ONEFLOW_CORE_GRAPH_OUT_BOXING_TASK_NODE_H_
#define ONEFLOW_CORE_GRAPH_OUT_BOXING_TASK_NODE_H_

#include "oneflow/core/graph/boxing_task_node.h"

namespace oneflow {

class OutBoxingTaskNode final : public BoxingTaskNode {
 public:
  OF_DISALLOW_COPY_AND_MOVE(OutBoxingTaskNode);
  OutBoxingTaskNode() = default;
  ~OutBoxingTaskNode() = default;

 private:
  std::unique_ptr<TaskNode> CreateSameTypeNode() const override {
    return of_make_unique<OutBoxingTaskNode>();
  }
  void InitWithFwNode(TaskNode* fw_node) override {
    BoxingTaskNode::InitWithFwNode(fw_node);
  }
  void FwVirtualBuild() override;
};

}  // namespace oneflow

#endif  // ONEFLOW_CORE_GRAPH_OUT_BOXING_TASK_NODE_H_