Home
last modified time | relevance | path

Searched defs:kernelHeight (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/test/
H A Dconvolution-operator-tester.h200 uint32_t kernelHeight, in kernelSize()
211 uint32_t kernelHeight, in kernelSize()
228 inline ConvolutionOperatorTester& kernelHeight(uint32_t kernelHeight) { in kernelHeight() argument
234 inline uint32_t kernelHeight() const { in kernelHeight() function
H A Ddeconvolution-operator-tester.h168 uint32_t kernelHeight, in kernelSize()
177 inline DeconvolutionOperatorTester& kernelHeight(uint32_t kernelHeight) { in kernelHeight() function
183 inline uint32_t kernelHeight() const { in kernelHeight() function
H A Ddwconv-microkernel-tester.h72 inline DWConvMicrokernelTester& kernelHeight(uint32_t kernelHeight) { in kernelHeight() argument
78 inline uint32_t kernelHeight() const { in kernelHeight() function
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DConv2dTestImpl.cpp228 unsigned int kernelHeight = armnn::numeric_cast<unsigned int>(originalKernelShape[2]); in SimpleConvolution2dTestImpl() local
403 unsigned int kernelHeight = armnn::numeric_cast<unsigned int>(kernelShape[1]); in SimpleConvolution2dNhwcTestImpl() local
1367 unsigned int kernelHeight = 3; in CompareConvolution2dTestImpl() local
1715 unsigned int kernelHeight = armnn::numeric_cast<unsigned int>(kernelShape[1]); in DepthwiseConvolution2dAsymmetricTestImpl() local
1865 unsigned int kernelHeight = 3; in DepthwiseConvolution2dDepthMul1TestImpl() local
2038 unsigned int kernelHeight = 5; in DepthwiseConvolution2dTestImpl() local
2272 unsigned int kernelHeight = armnn::numeric_cast<unsigned int>(originalKernelShape[1]); in DepthwiseConvolution2dTestImpl() local
3043 unsigned int kernelHeight = 3; in CompareDepthwiseConvolution2dTestImpl() local
H A DConv3dTestImpl.cpp185 unsigned int kernelHeight = armnn::numeric_cast<unsigned int>(kernelShape[1]); in SimpleConvolution3dTestImpl() local
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/
H A DDepthwiseConvolution2dEndToEndTests.hpp66 unsigned int kernelHeight = 5; in DepthwiseConvolution2dEndToEnd() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/bench/
H A Dconvolution.cc26 const size_t kernelHeight = state.range(3); in convolution_q8() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DDepthwiseConv2d.cu293 const int kernelWidth, const int kernelHeight, in conv_depthwise2d_grad_weight_kernel()