LstmVisitor Class Reference

#include <ConstTensorLayerVisitor.hpp>

Inheritance diagram for LstmVisitor:

Public Member Functions
	LstmVisitor (const LstmInputParams &params, const char *name=nullptr)
Public Member Functions inherited from TestLayerVisitor
	TestLayerVisitor (const char *name)
Public Member Functions inherited from StrategyBase< NoThrowStrategy >
virtual 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 IStrategy
virtual void	FinishStrategy ()

Protected Member Functions
template<typename LayerType >
void	CheckInputParameters (const LayerType *layer, const LstmInputParams &inputParams)
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 ()

Protected Attributes
LstmInputParams	m_InputParams
Protected Attributes inherited from StrategyBase< NoThrowStrategy >
NoThrowStrategy	m_DefaultStrategy

Detailed Description

Definition at line 258 of file ConstTensorLayerVisitor.hpp.

Constructor & Destructor Documentation

LstmVisitor()

LstmVisitor ( const LstmInputParams &  params, const char *  name = nullptr  )

inlineexplicit

Definition at line 261 of file ConstTensorLayerVisitor.hpp.

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

Member Function Documentation

CheckInputParameters()

void CheckInputParameters ( const LayerType *  layer, const LstmInputParams &  inputParams  )

protected

Definition at line 274 of file ConstTensorLayerVisitor.hpp.

References TestLayerVisitor::CheckConstTensorPtrs(), LstmInputParams::m_CellBias, LstmInputParams::m_CellLayerNormWeights, LstmInputParams::m_CellToForgetWeights, LstmInputParams::m_CellToInputWeights, LstmInputParams::m_CellToOutputWeights, LstmInputParams::m_ForgetGateBias, LstmInputParams::m_ForgetLayerNormWeights, LstmInputParams::m_InputGateBias, LstmInputParams::m_InputLayerNormWeights, LstmInputParams::m_InputToCellWeights, LstmInputParams::m_InputToForgetWeights, LstmInputParams::m_InputToInputWeights, LstmInputParams::m_InputToOutputWeights, LstmInputParams::m_OutputGateBias, LstmInputParams::m_OutputLayerNormWeights, LstmInputParams::m_ProjectionBias, LstmInputParams::m_ProjectionWeights, LstmInputParams::m_RecurrentToCellWeights, LstmInputParams::m_RecurrentToForgetWeights, LstmInputParams::m_RecurrentToInputWeights, and LstmInputParams::m_RecurrentToOutputWeights.

{
    CheckConstTensorPtrs("OutputGateBias",
                         inputParams.m_OutputGateBias,
                         layer->m_BasicParameters.m_OutputGateBias);
    CheckConstTensorPtrs("InputToForgetWeights",
                         inputParams.m_InputToForgetWeights,
                         layer->m_BasicParameters.m_InputToForgetWeights);
    CheckConstTensorPtrs("InputToCellWeights",
                         inputParams.m_InputToCellWeights,
                         layer->m_BasicParameters.m_InputToCellWeights);
    CheckConstTensorPtrs("InputToOutputWeights",
                         inputParams.m_InputToOutputWeights,
                         layer->m_BasicParameters.m_InputToOutputWeights);
    CheckConstTensorPtrs("RecurrentToForgetWeights",
                         inputParams.m_RecurrentToForgetWeights,
                         layer->m_BasicParameters.m_RecurrentToForgetWeights);
    CheckConstTensorPtrs("RecurrentToCellWeights",
                         inputParams.m_RecurrentToCellWeights,
                         layer->m_BasicParameters.m_RecurrentToCellWeights);
    CheckConstTensorPtrs("RecurrentToOutputWeights",
                         inputParams.m_RecurrentToOutputWeights,
                         layer->m_BasicParameters.m_RecurrentToOutputWeights);
    CheckConstTensorPtrs("ForgetGateBias",
                         inputParams.m_ForgetGateBias,
                         layer->m_BasicParameters.m_ForgetGateBias);
    CheckConstTensorPtrs("CellBias",
                         inputParams.m_CellBias,
                         layer->m_BasicParameters.m_CellBias);

    CheckConstTensorPtrs("InputToInputWeights",
                         inputParams.m_InputToInputWeights,
                         layer->m_CifgParameters.m_InputToInputWeights);
    CheckConstTensorPtrs("RecurrentToInputWeights",
                         inputParams.m_RecurrentToInputWeights,
                         layer->m_CifgParameters.m_RecurrentToInputWeights);
    CheckConstTensorPtrs("InputGateBias",
                         inputParams.m_InputGateBias,
                         layer->m_CifgParameters.m_InputGateBias);

    CheckConstTensorPtrs("ProjectionBias",
                         inputParams.m_ProjectionBias,
                         layer->m_ProjectionParameters.m_ProjectionBias);
    CheckConstTensorPtrs("ProjectionWeights",
                         inputParams.m_ProjectionWeights,
                         layer->m_ProjectionParameters.m_ProjectionWeights);

    CheckConstTensorPtrs("CellToInputWeights",
                         inputParams.m_CellToInputWeights,
                         layer->m_PeepholeParameters.m_CellToInputWeights);
    CheckConstTensorPtrs("CellToForgetWeights",
                         inputParams.m_CellToForgetWeights,
                         layer->m_PeepholeParameters.m_CellToForgetWeights);
    CheckConstTensorPtrs("CellToOutputWeights",
                         inputParams.m_CellToOutputWeights,
                         layer->m_PeepholeParameters.m_CellToOutputWeights);

    CheckConstTensorPtrs("InputLayerNormWeights",
                         inputParams.m_InputLayerNormWeights,
                         layer->m_LayerNormParameters.m_InputLayerNormWeights);
    CheckConstTensorPtrs("ForgetLayerNormWeights",
                         inputParams.m_ForgetLayerNormWeights,
                         layer->m_LayerNormParameters.m_ForgetLayerNormWeights);
    CheckConstTensorPtrs("CellLayerNormWeights",
                         inputParams.m_CellLayerNormWeights,
                         layer->m_LayerNormParameters.m_CellLayerNormWeights);
    CheckConstTensorPtrs("OutputLayerNormWeights",
                         inputParams.m_OutputLayerNormWeights,
                         layer->m_LayerNormParameters.m_OutputLayerNormWeights);
}

Member Data Documentation

m_InputParams

LstmInputParams m_InputParams

protected

Definition at line 270 of file ConstTensorLayerVisitor.hpp.

---

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

• src/armnn/test/ConstTensorLayerVisitor.hpp