SampleDynamicLayerSupport Class Reference

#include <SampleDynamicLayerSupport.hpp>

Inheritance diagram for SampleDynamicLayerSupport:

Collaboration diagram for SampleDynamicLayerSupport:

Public Member Functions
bool	IsAdditionSupported (const armnn::TensorInfo &input0, const armnn::TensorInfo &input1, const armnn::TensorInfo &output, armnn::Optional< std::string & > reasonIfUnsupported=armnn::EmptyOptional()) const
bool	IsInputSupported (const armnn::TensorInfo &input, armnn::Optional< std::string & > reasonIfUnsupported) const
bool	IsOutputSupported (const armnn::TensorInfo &output, armnn::Optional< std::string & > reasonIfUnsupported) const
bool	IsLayerSupported (const armnn::LayerType &type, const std::vector< armnn::TensorInfo > &infos, const armnn::BaseDescriptor &descriptor, const armnn::Optional< armnn::LstmInputParamsInfo > &lstmParamsInfo, const armnn::Optional< armnn::QuantizedLstmInputParamsInfo > &quantizedLstmParamsInfo, armnn::Optional< std::string & > reasonIfUnsupported=armnn::EmptyOptional()) const
	Default implementation of the ILayerSupport interface, Backends should implement this as a switch statement for each of their LayerTypes calling their specific backend implementation of IsXXXLayerSupported. More...
Public Member Functions inherited from LayerSupportBase
bool	IsDetectionPostProcessSupported (const TensorInfo &boxEncodings, const TensorInfo &scores, const TensorInfo &anchors, const TensorInfo &detectionBoxes, const TensorInfo &detectionClasses, const TensorInfo &detectionScores, const TensorInfo &numDetections, const DetectionPostProcessDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
bool	IsMemCopySupported (const TensorInfo &input, const TensorInfo &output, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
bool	IsMemImportSupported (const TensorInfo &input, const TensorInfo &output, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
bool	IsMergeSupported (const TensorInfo &input0, const TensorInfo &input1, const TensorInfo &output, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
bool	IsQuantizedLstmSupported (const TensorInfo &input, const TensorInfo &previousCellStateIn, const TensorInfo &previousOutputIn, const TensorInfo &cellStateOut, const TensorInfo &output, const QuantizedLstmInputParamsInfo &paramsInfo, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
bool	IsShapeSupported (const TensorInfo &input, const TensorInfo &output, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const
bool	IsStandInSupported (const std::vector< const TensorInfo * > &inputs, const std::vector< const TensorInfo * > &outputs, const StandInDescriptor &descriptor, Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const

Additional Inherited Members
Protected Member Functions inherited from ILayerSupport
	ILayerSupport ()
virtual	~ILayerSupport ()

Detailed Description

Definition at line 13 of file SampleDynamicLayerSupport.hpp.

Member Function Documentation

IsAdditionSupported()

bool IsAdditionSupported	(	const armnn::TensorInfo &	input0,
		const armnn::TensorInfo &	input1,
		const armnn::TensorInfo &	output,
		armnn::Optional< std::string & >	reasonIfUnsupported = armnn::EmptyOptional()
	)		const

Definition at line 52 of file SampleDynamicLayerSupport.cpp.

{
 
    if (input0.GetDataType() != armnn::DataType::Float32)
    {
        return false;
    }
 
    if (input0.GetDataType() != input1.GetDataType())
    {
        return false;
    }
 
    if (input0.GetDataType() != output.GetDataType())
    {
        return false;
    }
 
    return true;
}

References armnn::Float32, and TensorInfo::GetDataType().

Referenced by SampleDynamicLayerSupport::IsLayerSupported().

IsInputSupported()

bool IsInputSupported	(	const armnn::TensorInfo &	input,
		armnn::Optional< std::string & >	reasonIfUnsupported
	)		const

Definition at line 40 of file SampleDynamicLayerSupport.cpp.

{
    return true;
}

Referenced by SampleDynamicLayerSupport::IsLayerSupported().

IsLayerSupported()

bool IsLayerSupported ( const armnn::LayerType &  type, const std::vector< armnn::TensorInfo > &  infos, const armnn::BaseDescriptor &  descriptor, const armnn::Optional< armnn::LstmInputParamsInfo > &  lstmParamsInfo, const armnn::Optional< armnn::QuantizedLstmInputParamsInfo > &  quantizedLstmParamsInfo, armnn::Optional< std::string & >  reasonIfUnsupported = armnn::EmptyOptional()  ) const

virtual

Default implementation of the ILayerSupport interface, Backends should implement this as a switch statement for each of their LayerTypes calling their specific backend implementation of IsXXXLayerSupported.

Reimplemented from LayerSupportBase.

Definition at line 15 of file SampleDynamicLayerSupport.cpp.

{
    switch (type)
    {
        case armnn::LayerType::Input:
            return IsInputSupported(infos[0], reasonIfUnsupported);
        case armnn::LayerType::Output:
            return IsOutputSupported(infos[0], reasonIfUnsupported);
        case armnn::LayerType::Addition:
            return IsAdditionSupported(infos[0],
                                       infos[1],
                                       infos[2],
                                       reasonIfUnsupported);
        default:
            return false;
    }
}

References armnn::Addition, armnn::Input, SampleDynamicLayerSupport::IsAdditionSupported(), SampleDynamicLayerSupport::IsInputSupported(), SampleDynamicLayerSupport::IsOutputSupported(), and armnn::Output.

IsOutputSupported()

bool IsOutputSupported	(	const armnn::TensorInfo &	output,
		armnn::Optional< std::string & >	reasonIfUnsupported
	)		const

Definition at line 46 of file SampleDynamicLayerSupport.cpp.

{
    return true;
}

Referenced by SampleDynamicLayerSupport::IsLayerSupported().

---

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

• src/dynamic/sample/SampleDynamicLayerSupport.hpp
• src/dynamic/sample/SampleDynamicLayerSupport.cpp