24.02
|
#include <LayerSupportBase.hpp>
Public Member Functions | |
bool | IsLayerSupported (const LayerType &type, const std::vector< TensorInfo > &infos, const BaseDescriptor &descriptor, const Optional< LstmInputParamsInfo > &lstmParamsInfo=EmptyOptional(), const Optional< QuantizedLstmInputParamsInfo > &quantizedLstmParamsInfo=EmptyOptional(), Optional< std::string & > reasonIfUnsupported=EmptyOptional()) const override |
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... | |
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 ¶msInfo, 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 () |
Definition at line 13 of file LayerSupportBase.hpp.
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 |
Definition at line 85 of file LayerSupportBase.cpp.
Referenced by armnn::IsLayerTypeSupported().
|
overridevirtual |
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 ILayerSupport.
Reimplemented in NeonLayerSupport, ClLayerSupport, RefLayerSupport, and SampleDynamicLayerSupport.
Definition at line 40 of file LayerSupportBase.cpp.
References LayerSupportBase::IsMemCopySupported(), LayerSupportBase::IsMemImportSupported(), LayerSupportBase::IsStandInSupported(), armnn::MemCopy, armnn::MemImport, and armnn::StandIn.
bool IsMemCopySupported | ( | const TensorInfo & | input, |
const TensorInfo & | output, | ||
Optional< std::string & > | reasonIfUnsupported = EmptyOptional() |
||
) | const |
Definition at line 98 of file LayerSupportBase.cpp.
Referenced by LayerSupportBase::IsLayerSupported(), ClLayerSupport::IsLayerSupported(), and armnn::IsLayerTypeSupported().
bool IsMemImportSupported | ( | const TensorInfo & | input, |
const TensorInfo & | output, | ||
Optional< std::string & > | reasonIfUnsupported = EmptyOptional() |
||
) | const |
Definition at line 105 of file LayerSupportBase.cpp.
Referenced by RefLayerSupport::IsLayerSupported(), LayerSupportBase::IsLayerSupported(), ClLayerSupport::IsLayerSupported(), and armnn::IsLayerTypeSupported().
bool IsMergeSupported | ( | const TensorInfo & | input0, |
const TensorInfo & | input1, | ||
const TensorInfo & | output, | ||
Optional< std::string & > | reasonIfUnsupported = EmptyOptional() |
||
) | const |
Definition at line 112 of file LayerSupportBase.cpp.
Referenced by RefLayerSupport::IsLayerSupported(), ClLayerSupport::IsLayerSupported(), and armnn::IsLayerTypeSupported().
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 |
Definition at line 120 of file LayerSupportBase.cpp.
Referenced by RefLayerSupport::IsLayerSupported().
bool IsShapeSupported | ( | const TensorInfo & | input, |
const TensorInfo & | output, | ||
Optional< std::string & > | reasonIfUnsupported = EmptyOptional() |
||
) | const |
Definition at line 131 of file LayerSupportBase.cpp.
Referenced by ClLayerSupport::IsLayerSupported(), and armnn::IsLayerTypeSupported().
bool IsStandInSupported | ( | const std::vector< const TensorInfo * > & | inputs, |
const std::vector< const TensorInfo * > & | outputs, | ||
const StandInDescriptor & | descriptor, | ||
Optional< std::string & > | reasonIfUnsupported = EmptyOptional() |
||
) | const |
Definition at line 138 of file LayerSupportBase.cpp.
References OptionalReferenceSwitch< std::is_reference< T >::value, T >::value().
Referenced by LayerSupportBase::IsLayerSupported().