#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.
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 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 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, IsAdditionSupported(), IsInputSupported(), 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 IsLayerSupported().

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

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

Public Member Functions

Additional Inherited Members

Detailed Description

Member Function Documentation

◆ IsAdditionSupported()

◆ IsInputSupported()

◆ IsLayerSupported()

◆ IsOutputSupported()