|
25.02
|
#include "NeonLayerSupport.hpp"#include "NeonBackendModelContext.hpp"#include <armnn/Exceptions.hpp>#include <armnn/Tensor.hpp>#include <armnn/Types.hpp>#include <armnn/BackendRegistry.hpp>#include <LayerSupportCommon.hpp>#include <armnn/utility/IgnoreUnused.hpp>#include <armnn/utility/NumericCast.hpp>#include <armnn/utility/PolymorphicDowncast.hpp>#include <aclCommon/ArmComputeUtils.hpp>#include <aclCommon/ArmComputeTensorUtils.hpp>#include <backendsCommon/WorkloadUtils.hpp>#include "workloads/NeonAbsWorkload.hpp"#include "workloads/NeonAdditionWorkload.hpp"#include "workloads/NeonActivationWorkload.hpp"#include "workloads/NeonArgMinMaxWorkload.hpp"#include "workloads/NeonBatchMatMulWorkload.hpp"#include "workloads/NeonBatchNormalizationWorkload.hpp"#include "workloads/NeonBatchToSpaceNdWorkload.hpp"#include "workloads/NeonCastWorkload.hpp"#include "workloads/NeonChannelShuffleWorkload.hpp"#include "workloads/NeonComparisonWorkload.hpp"#include "workloads/NeonConcatWorkload.hpp"#include "workloads/NeonConstantWorkload.hpp"#include "workloads/NeonConvertFp16ToFp32Workload.hpp"#include "workloads/NeonConvertFp32ToFp16Workload.hpp"#include "workloads/NeonConvolution2dWorkload.hpp"#include "workloads/NeonConvolution3dWorkload.hpp"#include "workloads/NeonDepthToSpaceWorkload.hpp"#include "workloads/NeonDepthwiseConvolutionWorkload.hpp"#include "workloads/NeonDequantizeWorkload.hpp"#include "workloads/NeonDivisionWorkload.hpp"#include "workloads/NeonElementwiseBinaryWorkload.hpp"#include "workloads/NeonExpWorkload.hpp"#include "workloads/NeonFloorDivWorkload.hpp"#include "workloads/NeonFullyConnectedWorkload.hpp"#include "workloads/NeonFusedWorkload.hpp"#include "workloads/NeonGatherWorkload.hpp"#include "workloads/NeonGatherNdWorkload.hpp"#include "workloads/NeonInstanceNormalizationWorkload.hpp"#include "workloads/NeonL2NormalizationFloatWorkload.hpp"#include "workloads/NeonLogWorkload.hpp"#include "workloads/NeonLogSoftmaxWorkload.hpp"#include "workloads/NeonLogicalAndWorkload.hpp"#include "workloads/NeonLogicalNotWorkload.hpp"#include "workloads/NeonLogicalOrWorkload.hpp"#include "workloads/NeonLstmFloatWorkload.hpp"#include "workloads/NeonMaximumWorkload.hpp"#include "workloads/NeonMeanWorkload.hpp"#include "workloads/NeonMinimumWorkload.hpp"#include "workloads/NeonMultiplicationWorkload.hpp"#include "workloads/NeonNegWorkload.hpp"#include "workloads/NeonNormalizationFloatWorkload.hpp"#include "workloads/NeonPadWorkload.hpp"#include "workloads/NeonPermuteWorkload.hpp"#include "workloads/NeonPooling2dWorkload.hpp"#include "workloads/NeonPooling3dWorkload.hpp"#include "workloads/NeonPreluWorkload.hpp"#include "workloads/NeonQLstmWorkload.hpp"#include "workloads/NeonQuantizeWorkload.hpp"#include "workloads/NeonQuantizedLstmWorkload.hpp"#include "workloads/NeonReduceWorkload.hpp"#include "workloads/NeonReshapeWorkload.hpp"#include "workloads/NeonResizeWorkload.hpp"#include "workloads/NeonReverseV2Workload.hpp"#include "workloads/NeonRsqrtWorkload.hpp"#include "workloads/NeonSinWorkload.hpp"#include "workloads/NeonSliceWorkload.hpp"#include "workloads/NeonSoftmaxWorkload.hpp"#include "workloads/NeonSpaceToBatchNdWorkload.hpp"#include "workloads/NeonSpaceToDepthWorkload.hpp"#include "workloads/NeonSplitterWorkload.hpp"#include "workloads/NeonSqrtWorkload.hpp"#include "workloads/NeonStackWorkload.hpp"#include "workloads/NeonStridedSliceWorkload.hpp"#include "workloads/NeonSubtractionWorkload.hpp"#include "workloads/NeonTileWorkload.hpp"#include "workloads/NeonTransposeConvolution2dWorkload.hpp"#include "workloads/NeonTransposeWorkload.hpp"#include "workloads/NeonUnidirectionalSequenceLstmFloatWorkload.hpp"#include "workloads/NeonUnidirectionalSequenceLstmWorkload.hpp"Go to the source code of this file.
Namespaces | |
| armnn | |
| Copyright (c) 2021 ARM Limited and Contributors. | |
Macros | |
| #define | FORWARD_WORKLOAD_VALIDATE_FUNC(func, reasonIfUnsupported, ...) return IsWorkloadSupported(func, reasonIfUnsupported, __VA_ARGS__); |
Functions | |
| bool | IsLayerTypeSupported (const LayerType &type, const std::vector< TensorInfo > &infos, const BaseDescriptor &descriptor, const Optional< LstmInputParamsInfo > &lstmParamsInfo, const Optional< QuantizedLstmInputParamsInfo > &quantizedLstmParamsInfo, Optional< std::string & > reasonIfUnsupported, const NeonLayerSupport &support) |
| #define FORWARD_WORKLOAD_VALIDATE_FUNC | ( | func, | |
| reasonIfUnsupported, | |||
| ... | |||
| ) | return IsWorkloadSupported(func, reasonIfUnsupported, __VA_ARGS__); |
Definition at line 167 of file NeonLayerSupport.cpp.