Home
last modified time | relevance | path

Searched defs:kernelShape (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/armnn/src/armnn/layers/
H A DTransposeConvolution2dLayer.cpp62 const TensorShape& kernelShape = inputShapes[1]; in InferOutputShapes() local
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DConv2dTestImpl.cpp383 const armnn::TensorShape& kernelShape, in SimpleConvolution2dNhwcTestImpl()
1387 unsigned int kernelShape[] = {outputChannels, inputChannels, kernelHeight, kernelWidth}; in CompareConvolution2dTestImpl() local
1699 const armnn::TensorShape& kernelShape, in DepthwiseConvolution2dAsymmetricTestImpl()
3064 std::vector<unsigned int> kernelShape{ 1, kernelHeight, kernelWidth, outputChannels }; in CompareDepthwiseConvolution2dTestImpl() local
H A DConv3dTestImpl.cpp159 const armnn::TensorShape& kernelShape, in SimpleConvolution3dTestImpl()