diff options
Diffstat (limited to 'include/armnn/backends/OptimizationViews.hpp')
-rw-r--r-- | include/armnn/backends/OptimizationViews.hpp | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/include/armnn/backends/OptimizationViews.hpp b/include/armnn/backends/OptimizationViews.hpp index c357c0c8bb..f3479febd3 100644 --- a/include/armnn/backends/OptimizationViews.hpp +++ b/include/armnn/backends/OptimizationViews.hpp @@ -56,8 +56,11 @@ public: bool Validate(const SubgraphView& originalSubgraph) const; + ARMNN_DEPRECATED_MSG_REMOVAL_DATE("GetGraph is deprecated. Use GetINetwork instead.", "22.08") Graph& GetGraph() { return m_Graph; } + INetworkPtr& GetINetwork() { return m_INetwork; } + private: Substitutions m_SuccesfulOptimizations; ///< Proposed substitutions from successful optimizations Subgraphs m_FailedOptimizations; ///< Subgraphs from the original subgraph which cannot be supported @@ -65,6 +68,11 @@ private: /// Graph object used only as a container for any layer generated by the optimization process Graph m_Graph; + + /// INetworkPtr object used only as a container for any layer generated by the optimization process + /// Also, can use to AddPrecompiledLayer to the SubstitutionPair + /// Use in favour of m_Graph which depreciates 22.08 + INetworkPtr m_INetwork = INetwork::Create();; }; } //namespace armnn |