Home
last modified time | relevance | path

Searched defs:outputDepth (Results 1 – 4 of 4) sorted by relevance

/external/armnn/src/backends/reference/workloads/
DTransposeConvolution2d.cpp45 const unsigned int outputDepth = outputShape[channelsIndex]; in TransposeConvolution2dImpl() local
DConv3dImpl.cpp47 const unsigned int outputDepth = rOutputShape[depthIndex]; in Convolve3d() local
/external/armnn/src/backends/backendsCommon/test/layerTests/
DPooling3dTestImpl.cpp68 unsigned int outputDepth = armnn::numeric_cast<unsigned int>(outputShape[depthIndex]); in SimplePooling3dTestImpl() local
177 unsigned int outputDepth = in SimpleMaxPooling3dSize2x2x2Stride1x1x1TestCommon() local
1139 const unsigned int outputDepth = (inputDepth + 2 * padZ + strideZ - poolSize) / strideZ; in ComparePooling3dTestCommon() local
DConv3dTestImpl.cpp191 unsigned int outputDepth = armnn::numeric_cast<unsigned int>(outputExpectedShape[1]); in SimpleConvolution3dTestImpl() local