Searched defs:outputDepth (Results 1 – 7 of 7) sorted by relevance
83 const int64_t outputDepth = in get_output_shape() local120 const int64_t outputDepth = output_shape[output_shape.size() - 3]; in q_avg_pool3d() local
45 const unsigned int outputDepth = outputShape[channelsIndex]; in TransposeConvolution2dImpl() local
47 const unsigned int outputDepth = rOutputShape[depthIndex]; in Convolve3d() local
68 unsigned int outputDepth = armnn::numeric_cast<unsigned int>(outputShape[depthIndex]); in SimplePooling3dTestImpl() local177 unsigned int outputDepth = in SimpleMaxPooling3dSize2x2x2Stride1x1x1TestCommon() local1139 const unsigned int outputDepth = (inputDepth + 2 * padZ + strideZ - poolSize) / strideZ; in ComparePooling3dTestCommon() local
191 unsigned int outputDepth = armnn::numeric_cast<unsigned int>(outputExpectedShape[1]); in SimpleConvolution3dTestImpl() local
412 inline size_t outputDepth() const { in outputDepth() function
2005 int64_t outputDepth,2165 int64_t outputDepth,