From c65023e01d8e5c8f5af15dd8e2ca4fae3c4889f9 Mon Sep 17 00:00:00 2001
From: Sheri Zhang <sheri.zhang@arm.com>
Date: Wed, 3 Nov 2021 21:24:00 +0000
Subject: Refine code for graph level fusion

Partially resolves: COMPMID-4701

Signed-off-by: Sheri Zhang <sheri.zhang@arm.com>
Change-Id: I11fb89bd31ef5a98ad9a2d6bb5fd0c1e8371b52f
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6572
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gunes Bayir <gunes.bayir@arm.com>
Reviewed-by: SiCong Li <sicong.li@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
---
 arm_compute/graph/INode.h | 28 ++++++++++++++++++++--------
 1 file changed, 20 insertions(+), 8 deletions(-)

(limited to 'arm_compute/graph/INode.h')

diff --git a/arm_compute/graph/INode.h b/arm_compute/graph/INode.h
index b92003464c..becd672d90 100644
--- a/arm_compute/graph/INode.h
+++ b/arm_compute/graph/INode.h
@@ -1,5 +1,5 @@
 /*
- * Copyright (c) 2018-2019 Arm Limited.
+ * Copyright (c) 2018-2019,2021 Arm Limited.
  *
  * SPDX-License-Identifier: MIT
  *
@@ -29,6 +29,7 @@
 #include "arm_compute/graph/TensorDescriptor.h"
 #include "arm_compute/graph/Types.h"
 
+#include <list>
 #include <set>
 
 namespace arm_compute
@@ -240,18 +241,29 @@ public:
      * @return Assigned target of this node
      */
     Target assigned_target() const;
+    /** Post operator info list
+     *
+     * @return Post operator info list
+     */
+    const std::list<std::unique_ptr<ConvPostOpInfo>> &post_op_info_list() const;
+    /** Post operator info list
+     *
+     * @return Post operator info list
+     */
+    std::list<std::unique_ptr<ConvPostOpInfo>> &post_op_info_list();
 
 protected:
     friend class Graph;
 
 protected:
-    Graph                *_graph;           /**< Backward reference to graph owning the node */
-    NodeID                _id;              /**< Node ID */
-    NodeParams            _common_params;   /**< Node common params */
-    std::vector<TensorID> _outputs;         /**< Output of the node */
-    std::vector<EdgeID>   _input_edges;     /**< Inputs edge set */
-    std::set<EdgeID>      _output_edges;    /**< Output edge set */
-    Target                _assigned_target; /**< Assigned target by the Graph executor */
+    Graph                                     *_graph;             /**< Backward reference to graph owning the node */
+    NodeID                                     _id;                /**< Node ID */
+    NodeParams                                 _common_params;     /**< Node common params */
+    std::vector<TensorID>                      _outputs;           /**< Output of the node */
+    std::vector<EdgeID>                        _input_edges;       /**< Inputs edge set */
+    std::set<EdgeID>                           _output_edges;      /**< Output edge set */
+    Target                                     _assigned_target;   /**< Assigned target by the Graph executor */
+    std::list<std::unique_ptr<ConvPostOpInfo>> _post_op_info_list; /**< Post operator info list */
 };
 } // namespace graph
 } // namespace arm_compute
-- 
cgit v1.2.1