FuseConvertFp32ToBf16IntoConstLayers Class Reference

#include <FuseConvertFp32ToBf16IntoConstLayers.hpp>

Public Member Functions
void	Run (Graph &graph, InputSlot &connection) const

Protected Member Functions
	FuseConvertFp32ToBf16IntoConstLayers ()=default
	~FuseConvertFp32ToBf16IntoConstLayers ()=default

Detailed Description

Definition at line 17 of file FuseConvertFp32ToBf16IntoConstLayers.hpp.

Constructor & Destructor Documentation

FuseConvertFp32ToBf16IntoConstLayers()

FuseConvertFp32ToBf16IntoConstLayers ( )

protecteddefault

Referenced by FuseConvertFp32ToBf16IntoConstLayers::Run().

~FuseConvertFp32ToBf16IntoConstLayers()

~FuseConvertFp32ToBf16IntoConstLayers ( )

protecteddefault

Referenced by FuseConvertFp32ToBf16IntoConstLayers::Run().

Member Function Documentation

Run()

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

inline

Definition at line 20 of file FuseConvertFp32ToBf16IntoConstLayers.hpp.

References ARMNN_ASSERT, armnn::Constant, FloatingPointConverter::ConvertFloat32ToBFloat16(), armnn::ConvertFp32ToBf16, armnn::Float32, FuseConvertFp32ToBf16IntoConstLayers::FuseConvertFp32ToBf16IntoConstLayers(), InputSlot::GetConnectedOutputSlot(), Layer::GetDataType(), armnn::GetDataTypeName(), TensorInfo::GetNumElements(), Layer::GetOutputSlot(), InputSlot::GetOwningLayer(), OutputSlot::GetOwningLayer(), OutputSlot::GetTensorInfo(), Layer::GetType(), armnn::IgnoreUnused(), TensorInfo::IsConstant(), ConstantLayer::m_LayerOutput, OutputSlot::MoveAllConnections(), TensorInfo::SetConstant(), OutputSlot::SetTensorInfo(), and FuseConvertFp32ToBf16IntoConstLayers::~FuseConvertFp32ToBf16IntoConstLayers().

    {
        Layer& base = connection.GetConnectedOutputSlot()->GetOwningLayer();
        Layer& child = connection.GetOwningLayer();

        ARMNN_ASSERT(base.GetType() == LayerType::Constant);
        ARMNN_ASSERT(child.GetType() == LayerType::ConvertFp32ToBf16);

        auto dataType = base.GetDataType();
        switch (dataType)
        {
            case DataType::Float32:
                ReplaceConvertFp32ToBf16Layer<DataType::BFloat16>(
                        graph,
                        PolymorphicDowncast<ConstantLayer*>(&base),
                        PolymorphicDowncast<ConvertFp32ToBf16Layer*>(&child));
                break;
            default:
                throw InvalidArgumentException(GetDataTypeName(dataType) +
                                               std::string(" Constant Layer cannot be fused into ")  +
                                               GetDataTypeName(child.GetDataType()) +
                                               std::string(" conversion layer."));
        }
    }

---

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

• src/armnn/optimizations/FuseConvertFp32ToBf16IntoConstLayers.hpp