Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A DAveragePool3d.cpp83 const int64_t outputDepth = in get_output_shape() local
120 const int64_t outputDepth = output_shape[output_shape.size() - 3]; in q_avg_pool3d() local
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DTransposeConvolution2d.cpp45 const unsigned int outputDepth = outputShape[channelsIndex]; in TransposeConvolution2dImpl() local
H A DConv3dImpl.cpp47 const unsigned int outputDepth = rOutputShape[depthIndex]; in Convolve3d() local
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A 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
H A DConv3dTestImpl.cpp191 unsigned int outputDepth = armnn::numeric_cast<unsigned int>(outputExpectedShape[1]); in SimpleConvolution3dTestImpl() local
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DConv2D.cpp368 uint32_t outputDepth = getSizeOfDimension(outputShape, 3); in convQuant8PerChannelNhwc() local
461 uint32_t outputDepth = getSizeOfDimension(outputShape, 3); in convQuant8PerChannelNhwc() local
DReshape.cpp125 const int outputDepth = extOutputShape.Dims(3); in padGeneric() local
DDepthwiseConv2D.cpp294 uint32_t outputDepth = getSizeOfDimension(outputShape, 3); in depthwiseConvQuant8PerChannelNhwc() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/test/
H A Dconvolution-operator-tester.h412 inline size_t outputDepth() const { in outputDepth() function
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/
H A DQuantizedOpKernels.cpp2005 int64_t outputDepth,
2165 int64_t outputDepth,