Home
last modified time | relevance | path

Searched refs:originalKernelShape (Results 1 – 1 of 1) sorted by relevance

/external/armnn/src/backends/backendsCommon/test/layerTests/
DConv2dTestImpl.cpp203 const armnn::TensorShape& originalKernelShape, in SimpleConvolution2dTestImpl() argument
228 unsigned int kernelHeight = armnn::numeric_cast<unsigned int>(originalKernelShape[2]); in SimpleConvolution2dTestImpl()
229 unsigned int kernelWidth = armnn::numeric_cast<unsigned int>(originalKernelShape[3]); in SimpleConvolution2dTestImpl()
230 unsigned int kernelChannels = armnn::numeric_cast<unsigned int>(originalKernelShape[1]); in SimpleConvolution2dTestImpl()
231 unsigned int kernelDepthMul = armnn::numeric_cast<unsigned int>(originalKernelShape[0]); in SimpleConvolution2dTestImpl()
2248 const armnn::TensorShape& originalKernelShape, in DepthwiseConvolution2dTestImpl() argument
2272 unsigned int kernelHeight = armnn::numeric_cast<unsigned int>(originalKernelShape[1]); in DepthwiseConvolution2dTestImpl()
2273 unsigned int kernelWidth = armnn::numeric_cast<unsigned int>(originalKernelShape[2]); in DepthwiseConvolution2dTestImpl()
2274 unsigned int kernelChannels = armnn::numeric_cast<unsigned int>(originalKernelShape[3]); in DepthwiseConvolution2dTestImpl()