ArmNN
 26.07
Public Member Functions | Protected Member Functions | List of all members
ConvertConstDequantisationLayersToConstLayersImpl Class Reference

#include <ConvertConstDequantisationLayersToConstLayers.hpp>

Public Member Functions

void Run (Graph &graph, InputSlot &connection) const
 

Protected Member Functions

 ConvertConstDequantisationLayersToConstLayersImpl ()=default
 
 ~ConvertConstDequantisationLayersToConstLayersImpl ()=default
 

Detailed Description

Definition at line 18 of file ConvertConstDequantisationLayersToConstLayers.hpp.

Constructor & Destructor Documentation

◆ ConvertConstDequantisationLayersToConstLayersImpl()

ConvertConstDequantisationLayersToConstLayersImpl ( )
protecteddefault

◆ ~ConvertConstDequantisationLayersToConstLayersImpl()

~ConvertConstDequantisationLayersToConstLayersImpl ( )
protecteddefault

References ARMNN_ASSERT, ARMNN_LOG, FloatingPointConverter::ConvertFloat16To32(), armnn::Convolution2d, armnn::Float16, armnn::Float32, InputSlot::GetConnectedOutputSlot(), OutputSlot::GetConnection(), TensorInfo::GetDataType(), armnn::GetDataTypeName(), armnn::GetDataTypeSize(), Layer::GetInputSlot(), OutputSlot::GetNumConnections(), TensorInfo::GetNumElements(), Layer::GetNumOutputSlots(), Layer::GetOutputSlot(), InputSlot::GetOwningLayer(), TensorInfo::GetQuantizationOffset(), TensorInfo::GetQuantizationScale(), TensorInfo::GetShape(), OutputSlot::GetTensorInfo(), armnn::info, TensorInfo::IsConstant(), ConstantLayer::m_LayerOutput, OutputSlot::MoveAllConnections(), armnn::NCHW, armnnUtils::Permute(), armnn::QAsymmS8, armnn::QSymmS8, TensorInfo::SetConstant(), and OutputSlot::SetTensorInfo().

Member Function Documentation

◆ Run()

void Run ( Graph &  graph,
InputSlot &  connection 
) const
inline

Definition at line 21 of file ConvertConstDequantisationLayersToConstLayers.hpp.

22  {
23  Layer& base = connection.GetConnectedOutputSlot()->GetOwningLayer();
24  Layer& child = connection.GetOwningLayer();
25 
26  ARMNN_ASSERT(base.GetType() == LayerType::Constant);
27  ARMNN_ASSERT(child.GetType() == LayerType::Dequantize);
28 
29  ReplaceConstDequantisationLayer(graph,
30  PolymorphicDowncast<ConstantLayer*>(&base),
31  PolymorphicDowncast<DequantizeLayer*>(&child));
32 
33  }
ARMNN_ASSERT
#define ARMNN_ASSERT(COND)
Definition: Assert.hpp:14

References ARMNN_ASSERT, armnn::Constant, armnn::Dequantize, InputSlot::GetConnectedOutputSlot(), InputSlot::GetOwningLayer(), OutputSlot::GetOwningLayer(), and Layer::GetType().

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