21.11
|
#include <ConstTensorLayerVisitor.hpp>
Public Member Functions | |
TestLstmLayerVisitor (const LstmDescriptor &descriptor, const LstmInputParams ¶ms, 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 |
![]() | |
LstmVisitor (const LstmInputParams ¶ms, const char *name=nullptr) | |
![]() | |
TestLayerVisitor (const char *name) | |
![]() | |
virtual void | FinishStrategy () |
Protected Member Functions | |
void | CheckDescriptor (const LstmDescriptor &descriptor) |
![]() | |
template<typename LayerType > | |
void | CheckInputParameters (const LayerType *layer, const LstmInputParams &inputParams) |
![]() | |
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) |
![]() | |
virtual | ~StrategyBase () |
![]() | |
IStrategy () | |
virtual | ~IStrategy () |
Additional Inherited Members | |
![]() | |
LstmInputParams | m_InputParams |
![]() | |
NoThrowStrategy | m_DefaultStrategy |
Definition at line 345 of file ConstTensorLayerVisitor.hpp.
|
inlineexplicit |
Definition at line 348 of file ConstTensorLayerVisitor.hpp.
|
protected |
Definition at line 51 of file ConstTensorLayerVisitor.cpp.
References LstmDescriptor::m_ActivationFunc, LstmDescriptor::m_CifgEnabled, LstmDescriptor::m_ClippingThresCell, LstmDescriptor::m_ClippingThresProj, LstmDescriptor::m_PeepholeEnabled, and LstmDescriptor::m_ProjectionEnabled.
|
inlineoverridevirtual |
Reimplemented from StrategyBase< NoThrowStrategy >.
Definition at line 355 of file ConstTensorLayerVisitor.hpp.
References NoThrowStrategy::Apply(), TestConvolution2dLayerVisitor::CheckDescriptor(), TestLayerVisitor::CheckLayerName(), TestLayerVisitor::CheckLayerPointer(), armnn::GetLayerTypeAsCString(), IConnectableLayer::GetType(), armnn::IgnoreUnused(), armnn::Lstm, and StrategyBase< NoThrowStrategy >::m_DefaultStrategy.