diff options
| author | Sheri Zhang <sheri.zhang@arm.com> | 2021-11-03 21:24:00 +0000 |
|---|---|---|
| committer | Sheri Zhang <sheri.zhang@arm.com> | 2021-11-04 15:35:58 +0000 |
| commit | c65023e01d8e5c8f5af15dd8e2ca4fae3c4889f9 (patch) | |
| tree | 313c5c0786971986d92c2e59073eb06406f43990 /arm_compute/graph/INode.h | |
| parent | cc859158faff318a81dd9aafcc5b609d7dd5d4a9 (diff) | |
| download | ComputeLibrary-c65023e01d8e5c8f5af15dd8e2ca4fae3c4889f9.tar.gz | |
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>
Diffstat (limited to 'arm_compute/graph/INode.h')
| -rw-r--r-- | arm_compute/graph/INode.h | 28 |
1 files changed, 20 insertions, 8 deletions
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 |