ArmNN
 22.05.01
AddDebugImpl Class Reference

#include <AddDebug.hpp>

Public Member Functions

void Run (Graph &graph, Layer &layer) const
 

Protected Member Functions

 AddDebugImpl ()=default
 
 ~AddDebugImpl ()=default
 

Detailed Description

Definition at line 15 of file AddDebug.hpp.

Constructor & Destructor Documentation

◆ AddDebugImpl()

AddDebugImpl ( )
protecteddefault

Referenced by AddDebugImpl::Run().

◆ ~AddDebugImpl()

~AddDebugImpl ( )
protecteddefault

Referenced by AddDebugImpl::Run().

Member Function Documentation

◆ Run()

void Run ( Graph graph,
Layer layer 
) const
inline

Definition at line 19 of file AddDebug.hpp.

References AddDebugImpl::AddDebugImpl(), armnn::Debug, Layer::GetType(), armnn::InsertDebugLayerAfter(), armnn::Output, and AddDebugImpl::~AddDebugImpl().

20  {
21  if (layer.GetType() != LayerType::Debug && layer.GetType() != LayerType::Output)
22  {
23  // if the inputs/outputs of this layer do not have a debug layer
24  // insert the debug layer after them
25  InsertDebugLayerAfter(graph, layer);
26  }
27  }
std::vector< DebugLayer * > InsertDebugLayerAfter(Graph &graph, Layer &layer)

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