1 files changed, 8 insertions, 12 deletions
diff --git a/runtimes/neurun/src/backend/acl_cl/StageGenerator.h b/runtimes/neurun/src/backend/acl_cl/StageGenerator.h
index 921604649..1dac2592b 100644
--- a/runtimes/neurun/src/backend/acl_cl/StageGenerator.h
+++ b/runtimes/neurun/src/backend/acl_cl/StageGenerator.h
@@ -17,9 +17,9 @@
 #ifndef __NEURUN_BACKEND_ACL_CL_STAGE_GENERATOR_H__
 #define __NEURUN_BACKEND_ACL_CL_STAGE_GENERATOR_H__
 
-#include "backend/IStageGenerator.h"
+#include "backend/interface/IStageGenerator.h"
 
-#include "graph/operand/Set.h"
+#include "model/operand/Set.h"
 #include "backend/acl_cl/TensorBuilder.h"
 
 namespace neurun
@@ -32,22 +32,18 @@ namespace acl_cl
 class StageGenerator : public IStageGenerator
 {
 public:
-  StageGenerator(const neurun::graph::operand::Set &ctx,
+  StageGenerator(const neurun::model::operand::Set &ctx,
                  const std::shared_ptr<TensorBuilder> &tensor_builder);
 
   virtual std::shared_ptr<ITensorBuilder> tensor_builder() override { return _tensor_builder; }
 
-  virtual Stage generate(const graph::operation::Conv2D::Implicit::Node &node) override;
-  virtual Stage generate(const graph::operation::MaxPool2D::Implicit::Node &node) override;
-  virtual Stage generate(const graph::operation::AvgPool2D::Implicit::Node &node) override;
-  virtual Stage generate(const graph::operation::Concat::Node &node) override;
-  virtual Stage generate(const graph::operation::FullyConnected::Node &node) override;
-  virtual Stage generate(const graph::operation::Reshape::Node &node) override;
-  virtual Stage generate(const graph::operation::Softmax::Node &node) override;
-  virtual Stage generate(const graph::operation::NOP::Node &node) override;
+#define OP(InternalName, IsNnApi, NnApiName) \
+  virtual void visit(const model::operation::InternalName &) override;
+#include "model/operation/Op.lst"
+#undef OP
 
 private:
-  const neurun::graph::operand::Set &_ctx;
+  const neurun::model::operand::Set &_ctx;
   std::shared_ptr<TensorBuilder> _tensor_builder;
 };