Lines Matching refs:arm_compute
27 inline arm_compute::NormalizationLayerInfo
57 …return arm_compute::NormalizationLayerInfo(arm_compute::NormType::CROSS_MAP, normSize, alpha, beta… in CreateAclNormalizationLayerInfoForL2Normalization()
60 inline arm_compute::ActivationLayerInfo::ActivationFunction
63 using AclActivationFunction = arm_compute::ActivationLayerInfo::ActivationFunction; in ConvertActivationFunctionToAclActivationFunction()
84 inline arm_compute::ActivationLayerInfo
87 …return arm_compute::ActivationLayerInfo(ConvertActivationFunctionToAclActivationFunction(actDesc.m… in ConvertActivationDescriptorToAclActivationLayerInfo()
91 inline arm_compute::ActivationLayerInfo
99 return arm_compute::ActivationLayerInfo(); in ConvertActivationDescriptorToAclActivationLayerInfo()
102 inline arm_compute::ActivationLayerInfo
112 return arm_compute::ActivationLayerInfo(); in ConvertAdditionalInfoToAclActivationLayerInfo()
115 inline arm_compute::ActivationLayerInfo
122 return arm_compute::ActivationLayerInfo(); // no activation, do nothing in ConvertLstmActivationFuncToAclLayerInfo()
124 …return arm_compute::ActivationLayerInfo(arm_compute::ActivationLayerInfo::ActivationFunction::RELU… in ConvertLstmActivationFuncToAclLayerInfo()
126 return arm_compute::ActivationLayerInfo( in ConvertLstmActivationFuncToAclLayerInfo()
127 arm_compute::ActivationLayerInfo::ActivationFunction::BOUNDED_RELU, 6.0); in ConvertLstmActivationFuncToAclLayerInfo()
129 return arm_compute::ActivationLayerInfo( in ConvertLstmActivationFuncToAclLayerInfo()
130 arm_compute::ActivationLayerInfo::ActivationFunction::TANH, 1.0, 1.0); in ConvertLstmActivationFuncToAclLayerInfo()
132 return arm_compute::ActivationLayerInfo( in ConvertLstmActivationFuncToAclLayerInfo()
133 arm_compute::ActivationLayerInfo::ActivationFunction::LOGISTIC); in ConvertLstmActivationFuncToAclLayerInfo()
139 inline arm_compute::ComparisonOperation ConvertComparisonOperationToAcl(const ComparisonDescriptor&… in ConvertComparisonOperationToAcl()
143 case ComparisonOperation::Greater: return arm_compute::ComparisonOperation::Greater; in ConvertComparisonOperationToAcl()
144 … case ComparisonOperation::GreaterOrEqual: return arm_compute::ComparisonOperation::GreaterEqual; in ConvertComparisonOperationToAcl()
145 case ComparisonOperation::Less: return arm_compute::ComparisonOperation::Less; in ConvertComparisonOperationToAcl()
146 … case ComparisonOperation::LessOrEqual: return arm_compute::ComparisonOperation::LessEqual; in ConvertComparisonOperationToAcl()
147 case ComparisonOperation::Equal: return arm_compute::ComparisonOperation::Equal; in ConvertComparisonOperationToAcl()
148 … case ComparisonOperation::NotEqual: return arm_compute::ComparisonOperation::NotEqual; in ConvertComparisonOperationToAcl()
153 inline arm_compute::PoolingType ConvertPoolingAlgorithmToAclPoolingType(PoolingAlgorithm poolingAlg… in ConvertPoolingAlgorithmToAclPoolingType()
155 using arm_compute::PoolingType; in ConvertPoolingAlgorithmToAclPoolingType()
166 inline arm_compute::DimensionRoundingType ConvertOutputShapeRoundingToAclDimensionRoundingType(Outp… in ConvertOutputShapeRoundingToAclDimensionRoundingType()
169 using arm_compute::DimensionRoundingType; in ConvertOutputShapeRoundingToAclDimensionRoundingType()
179 inline arm_compute::NormType
182 using arm_compute::NormType; in ConvertNormalizationAlgorithmChannelToAclNormType()
191 inline arm_compute::FullyConnectedLayerInfo
195 arm_compute::FullyConnectedLayerInfo fc_info; in ConvertFullyConnectedDescriptorToAclFullyConnectedLayerInfo()
201 inline arm_compute::FullyConnectedLayerInfo
203 arm_compute::ActivationLayerInfo activationLayerInfo) in ConvertFullyConnectedDescriptorToAclFullyConnectedLayerInfo()
205 arm_compute::FullyConnectedLayerInfo fc_info; in ConvertFullyConnectedDescriptorToAclFullyConnectedLayerInfo()
211 inline arm_compute::InterpolationPolicy ConvertResizeMethodToAclInterpolationPolicy(ResizeMethod re… in ConvertResizeMethodToAclInterpolationPolicy()
216 return arm_compute::InterpolationPolicy::BILINEAR; in ConvertResizeMethodToAclInterpolationPolicy()
218 return arm_compute::InterpolationPolicy::NEAREST_NEIGHBOR; in ConvertResizeMethodToAclInterpolationPolicy()
293 inline arm_compute::Conv3dInfo ComputeConv3DInfo(const armnn::Convolution3dDescriptor descriptor, in ComputeConv3DInfo()
297 …const arm_compute::Size3D stride{descriptor.m_StrideX, descriptor.m_StrideY, descriptor.m_Strid… in ComputeConv3DInfo()
298 const arm_compute::Padding3D padding{descriptor.m_PadLeft, descriptor.m_PadRight, in ComputeConv3DInfo()
301 …const arm_compute::Size3D dilation{descriptor.m_DilationX, descriptor.m_DilationY, descriptor.m… in ComputeConv3DInfo()
303 const arm_compute::ActivationLayerInfo activationInfo = in ComputeConv3DInfo()
305 const auto roundType = arm_compute::DimensionRoundingType::FLOOR; in ComputeConv3DInfo()
307 …return arm_compute::Conv3dInfo{stride, padding, activationInfo, dilation, roundType, isFastMathEna… in ComputeConv3DInfo()
310 inline arm_compute::Conv3dInfo ComputeConv3DInfo(const armnn::Convolution3dQueueDescriptor queueDes… in ComputeConv3DInfo()
314 …const arm_compute::Size3D stride{descriptor.m_StrideX, descriptor.m_StrideY, descriptor.m_Strid… in ComputeConv3DInfo()
315 const arm_compute::Padding3D padding{descriptor.m_PadLeft, descriptor.m_PadRight, in ComputeConv3DInfo()
318 …const arm_compute::Size3D dilation{descriptor.m_DilationX, descriptor.m_DilationY, descriptor.m… in ComputeConv3DInfo()
320 const arm_compute::ActivationLayerInfo activationInfo = in ComputeConv3DInfo()
322 const auto roundType = arm_compute::DimensionRoundingType::FLOOR; in ComputeConv3DInfo()
324 …return arm_compute::Conv3dInfo{stride, padding, activationInfo, dilation, roundType, isFastMathEna… in ComputeConv3DInfo()
327 inline arm_compute::PaddingMode ConvertPaddingModeToAcl(const PaddingMode& paddingMode) in ConvertPaddingModeToAcl()
331 case PaddingMode::Constant: return arm_compute::PaddingMode::CONSTANT; in ConvertPaddingModeToAcl()
332 case PaddingMode::Reflect: return arm_compute::PaddingMode::REFLECT; in ConvertPaddingModeToAcl()
333 case PaddingMode::Symmetric: return arm_compute::PaddingMode::SYMMETRIC; in ConvertPaddingModeToAcl()
338 inline arm_compute::ReductionOperation ConvertReductionOperationToAcl(const ReduceDescriptor& descr… in ConvertReductionOperationToAcl()
342 case ReduceOperation::Sum: return arm_compute::ReductionOperation::SUM; in ConvertReductionOperationToAcl()
343 case ReduceOperation::Mean: return arm_compute::ReductionOperation::MEAN_SUM; in ConvertReductionOperationToAcl()
344 case ReduceOperation::Max: return arm_compute::ReductionOperation::MAX; in ConvertReductionOperationToAcl()
345 case ReduceOperation::Min: return arm_compute::ReductionOperation::MIN; in ConvertReductionOperationToAcl()
346 case ReduceOperation::Prod: return arm_compute::ReductionOperation::PROD; in ConvertReductionOperationToAcl()