OptimizeForExclusiveConnectionImpl< BaseType, ChildType, Wrapped > Class Template Reference

Wrapper Optimization class that calls Wrapped::Run for every connection BaseType -> ChildType. More...

#include <Optimization.hpp>

Inheritance diagram for OptimizeForExclusiveConnectionImpl< BaseType, ChildType, Wrapped >:

Public Member Functions
void	Run (Graph &graph, BaseType &base) const

Protected Member Functions
	~OptimizeForExclusiveConnectionImpl ()=default

Detailed Description

template<typename BaseType, typename ChildType, typename Wrapped>
class armnn::OptimizeForExclusiveConnectionImpl< BaseType, ChildType, Wrapped >

Wrapper Optimization class that calls Wrapped::Run for every connection BaseType -> ChildType.

• Wrapped class mustn't remove the base layer. The optimizer will remove it if left unconnected after applying each optimization.
• Wrapped class mustn't affect existing connections in the same output. It might add new ones.
• Children layers are removed if left unconnected after applying the wrapped optimization.

Definition at line 131 of file Optimization.hpp.

Constructor & Destructor Documentation

~OptimizeForExclusiveConnectionImpl()

~OptimizeForExclusiveConnectionImpl ( )

protecteddefault

Member Function Documentation

Run()

void Run ( Graph &  graph, BaseType &  base  ) const

inline

Definition at line 136 of file Optimization.hpp.

References Graph::EraseLayer(), and Layer::IsOutputUnconnected().

    {
        for (auto output = base.BeginOutputSlots(); output != base.EndOutputSlots(); ++output)
        {
            if (output->GetNumConnections() == 1)
            {
                for (auto&& childInput : output->GetConnections())
                {
                    if (childInput->GetOwningLayer().GetType() == LayerEnumOf<ChildType>())
                    {
                        Wrapped::Run(graph, *childInput);
                    }
                }

                // Removes unconnected children.
                for (unsigned int i = 0; i < output->GetNumConnections();)
                {
                    Layer* child = &output->GetConnection(i)->GetOwningLayer();

                    if (child->IsOutputUnconnected())
                    {
                        graph.EraseLayer(child);
                    }
                    else
                    {
                        ++i;
                    }
                }
            }
        }
    }

---

The documentation for this class was generated from the following file:

• src/armnn/optimizations/Optimization.hpp