TestQuantizedLstmLayerVisitor Class Reference

#include <ConstTensorLayerVisitor.hpp>

Inheritance diagram for TestQuantizedLstmLayerVisitor:

Public Member Functions
	TestQuantizedLstmLayerVisitor (const QuantizedLstmInputParams &params, const char *name=nullptr)
void	ExecuteStrategy (const armnn::IConnectableLayer *layer, const armnn::BaseDescriptor &descriptor, const std::vector< armnn::ConstTensor > &constants, const char *name, const armnn::LayerBindingId id=0) override
Public Member Functions inherited from TestLayerVisitor
	TestLayerVisitor (const char *name)
Public Member Functions inherited from IStrategy
virtual void	FinishStrategy ()

Protected Member Functions
void	CheckInputParameters (const QuantizedLstmInputParams &params)
Protected Member Functions inherited from TestLayerVisitor
virtual	~TestLayerVisitor ()
void	CheckLayerName (const char *name)
void	CheckLayerPointer (const IConnectableLayer *layer)
void	CheckConstTensors (const ConstTensor &expected, const ConstTensor &actual)
void	CheckConstTensors (const ConstTensor &expected, const ConstTensorHandle &actual)
void	CheckConstTensorPtrs (const std::string &name, const ConstTensor *expected, const ConstTensor *actual)
void	CheckConstTensorPtrs (const std::string &name, const ConstTensor *expected, const std::shared_ptr< ConstTensorHandle > actual)
void	CheckOptionalConstTensors (const Optional< ConstTensor > &expected, const Optional< ConstTensor > &actual)
Protected Member Functions inherited from StrategyBase< NoThrowStrategy >
virtual	~StrategyBase ()
Protected Member Functions inherited from IStrategy
	IStrategy ()
virtual	~IStrategy ()

Additional Inherited Members
Protected Attributes inherited from StrategyBase< NoThrowStrategy >
NoThrowStrategy	m_DefaultStrategy

Detailed Description

Definition at line 404 of file ConstTensorLayerVisitor.hpp.

Constructor & Destructor Documentation

TestQuantizedLstmLayerVisitor()

TestQuantizedLstmLayerVisitor ( const QuantizedLstmInputParams &  params, const char *  name = nullptr  )

inlineexplicit

Definition at line 407 of file ConstTensorLayerVisitor.hpp.

        : TestLayerVisitor(name)
        , m_InputParams(params)
    {}

Member Function Documentation

CheckInputParameters()

void CheckInputParameters ( const QuantizedLstmInputParams &  params )

protected

Definition at line 70 of file ConstTensorLayerVisitor.cpp.

References TestLayerVisitor::CheckConstTensorPtrs(), QuantizedLstmInputParams::m_CellBias, QuantizedLstmInputParams::m_ForgetGateBias, QuantizedLstmInputParams::m_InputGateBias, QuantizedLstmInputParams::m_InputToCellWeights, QuantizedLstmInputParams::m_InputToForgetWeights, QuantizedLstmInputParams::m_InputToInputWeights, QuantizedLstmInputParams::m_InputToOutputWeights, QuantizedLstmInputParams::m_OutputGateBias, QuantizedLstmInputParams::m_RecurrentToCellWeights, QuantizedLstmInputParams::m_RecurrentToForgetWeights, QuantizedLstmInputParams::m_RecurrentToInputWeights, and QuantizedLstmInputParams::m_RecurrentToOutputWeights.

{
    CheckConstTensorPtrs("InputToInputWeights",
                         m_InputParams.m_InputToInputWeights,
                         inputParams.m_InputToInputWeights);

    CheckConstTensorPtrs("InputToForgetWeights",
                         m_InputParams.m_InputToForgetWeights,
                         inputParams.m_InputToForgetWeights);

    CheckConstTensorPtrs("InputToCellWeights",
                         m_InputParams.m_InputToCellWeights,
                         inputParams.m_InputToCellWeights);

    CheckConstTensorPtrs("InputToOutputWeights",
                         m_InputParams.m_InputToOutputWeights,
                         inputParams.m_InputToOutputWeights);

    CheckConstTensorPtrs("RecurrentToInputWeights",
                         m_InputParams.m_RecurrentToInputWeights,
                         inputParams.m_RecurrentToInputWeights);

    CheckConstTensorPtrs("RecurrentToForgetWeights",
                         m_InputParams.m_RecurrentToForgetWeights,
                         inputParams.m_RecurrentToForgetWeights);

    CheckConstTensorPtrs("RecurrentToCellWeights",
                         m_InputParams.m_RecurrentToCellWeights,
                         inputParams.m_RecurrentToCellWeights);

    CheckConstTensorPtrs("RecurrentToOutputWeights",
                         m_InputParams.m_RecurrentToOutputWeights,
                         inputParams.m_RecurrentToOutputWeights);

    CheckConstTensorPtrs("InputGateBias",  m_InputParams.m_InputGateBias,  inputParams.m_InputGateBias);
    CheckConstTensorPtrs("ForgetGateBias", m_InputParams.m_ForgetGateBias, inputParams.m_ForgetGateBias);
    CheckConstTensorPtrs("CellBias",       m_InputParams.m_CellBias,       inputParams.m_CellBias);
    CheckConstTensorPtrs("OutputGateBias", m_InputParams.m_OutputGateBias, inputParams.m_OutputGateBias);
}

ExecuteStrategy()

void ExecuteStrategy ( const armnn::IConnectableLayer *  layer, const armnn::BaseDescriptor &  descriptor, const std::vector< armnn::ConstTensor > &  constants, const char *  name, const armnn::LayerBindingId  id = 0  )

inlineoverridevirtual

Reimplemented from StrategyBase< NoThrowStrategy >.

Definition at line 413 of file ConstTensorLayerVisitor.hpp.

References NoThrowStrategy::Apply(), TestLayerVisitor::CheckLayerName(), TestLayerVisitor::CheckLayerPointer(), armnn::GetLayerTypeAsCString(), IConnectableLayer::GetType(), armnn::IgnoreUnused(), StrategyBase< NoThrowStrategy >::m_DefaultStrategy, and armnn::QuantizedLstm.

    {
        armnn::IgnoreUnused(descriptor, constants, id);
        switch (layer->GetType())
        {
            case armnn::LayerType::QuantizedLstm:
            {
                CheckLayerPointer(layer);
                CheckLayerName(name);
                CheckInputParameters(m_InputParams);
                break;
            }
            default:
            {
                m_DefaultStrategy.Apply(GetLayerTypeAsCString(layer->GetType()));
            }
        }
    }

---

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

• src/armnn/test/ConstTensorLayerVisitor.hpp
• src/armnn/test/ConstTensorLayerVisitor.cpp