Home
last modified time | relevance | path

Searched defs:depthMultiplier (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DDepthwiseConv2D.cpp134 int32_t depthMultiplier, int32_t activation, float* outputData, in depthwiseConvNhwc()
168 int32_t dilationHeightFactor, int32_t depthMultiplier, int32_t activation, in depthwiseConvNhwc()
194 int32_t depthMultiplier, int32_t activation, uint8_t* outputData, in depthwiseConvNhwc()
245 int32_t depthMultiplier, int32_t activation, int8_t* outputData, in depthwiseConvNhwc()
279 int32_t depthMultiplier, int32_t activation, T* outputData, const Shape& outputShape) { in depthwiseConvQuant8PerChannelNhwc()
374 int32_t depthMultiplier, int32_t activation, bool useNchw, T_Input* outputData, in depthwiseConv()
397 int32_t depthMultiplier, int32_t activation, bool useNchw, in depthwiseConvQuant8PerChannel()
/aosp_15_r20/external/armnn/src/backends/backendsCommon/
H A DWorkloadUtils.cpp145 unsigned int depthMultiplier = 1; in Convert1HWOTensorToAcl() local
217 unsigned int depthMultiplier = weightsShape[3] / inputInfo.GetShape()[channelIndex]; in Convert1HWOtoMIHW() local
/aosp_15_r20/external/armnn/src/backends/neon/workloads/
H A DNeonDepthwiseConvolutionWorkload.cpp110 unsigned int depthMultiplier = in NeonDepthwiseConvolutionWorkload() local
/aosp_15_r20/external/armnn/src/backends/cl/workloads/
H A DClDepthwiseConvolutionWorkload.cpp123 unsigned int depthMultiplier = in ClDepthwiseConvolutionWorkload() local
/aosp_15_r20/external/armnn/src/backends/neon/test/
H A DNeonLayerTests.cpp364 uint32_t depthMultiplier = 1, uint32_t padLeft = 0, uint32_t padRight = 0, in MakeDepthwiseConv2dDesc()
/aosp_15_r20/external/armnn/src/armnn/layers/
H A DDepthwiseConvolution2dLayer.cpp40 unsigned int depthMultiplier = filterShape[0]; in SerializeLayerParameters() local
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/
H A DDepthwiseConvolution2dEndToEndTests.hpp59 unsigned int depthMultiplier = 2; in DepthwiseConvolution2dEndToEnd() local
/aosp_15_r20/external/armnn/src/armnn/optimizations/
H A DFuseBatchNorm.hpp67 const unsigned int depthMultiplier = depthwise ? weightsShape[3] / inputChannels : 1; in Run() local
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DConvImpl.cpp103 const unsigned int depthMultiplier = depthwise ? outputChannels/inputChannels : 1; in Convolve() local
/aosp_15_r20/external/armnn/src/backends/aclCommon/
H A DArmComputeTensorUtils.cpp446 unsigned int depthMultiplier; in ComputeDepthwiseConv2dDepthMultiplier() local
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DConv2dTestImpl.cpp2031 unsigned int depthMultiplier = 2; in DepthwiseConvolution2dTestImpl() local