diff options
author | surmeh01 <surabhi.mehta@arm.com> | 2018-03-29 16:29:27 +0100 |
---|---|---|
committer | surmeh01 <surabhi.mehta@arm.com> | 2018-03-29 16:29:27 +0100 |
commit | bceff2fb3fc68bb0aa88b886900c34b77340c826 (patch) | |
tree | d867d3e090d58d3012dfbbac456e9ea8c7f789bc /src/armnn/Network.cpp | |
parent | 4fcda0101ec3d110c1d6d7bee5c83416b645528a (diff) | |
download | armnn-bceff2fb3fc68bb0aa88b886900c34b77340c826.tar.gz |
Release 18.03
Diffstat (limited to 'src/armnn/Network.cpp')
-rw-r--r-- | src/armnn/Network.cpp | 7 |
1 files changed, 6 insertions, 1 deletions
diff --git a/src/armnn/Network.cpp b/src/armnn/Network.cpp index 4ee68b3c48..77390cb0a4 100644 --- a/src/armnn/Network.cpp +++ b/src/armnn/Network.cpp @@ -58,6 +58,11 @@ Status OptimizedNetwork::PrintGraph() return Status::Success; } +Status OptimizedNetwork::SerializeToDot(std::ostream& stream) const +{ + return m_Graph->SerializeToDot(stream); +} + IOptimizedNetworkPtr Optimize(const INetwork& inNetwork, const DeviceSpec& deviceSpec) { const Network& network = *boost::polymorphic_downcast<const Network*>(&inNetwork); @@ -65,7 +70,7 @@ IOptimizedNetworkPtr Optimize(const INetwork& inNetwork, const DeviceSpec& devic OptimizedNetwork* optNet = new OptimizedNetwork(std::move(graph)); - Optimizer::Get().Optimize(optNet->GetGraph()); + Optimizer::Optimize(optNet->GetGraph()); // Infer the tensor infos for all output slots. Throws an exception on failure. optNet->GetGraph().InferTensorInfos(); |