Home
last modified time | relevance | path

Searched defs:outputBatchSize (Results 1 – 11 of 11) sorted by relevance

/aosp_15_r20/external/armnn/src/backends/neon/test/
H A DNeonTimerTest.cpp58 unsigned int outputBatchSize = inputBatchSize; variable
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DBatchToSpaceNd.cpp54 const unsigned int outputBatchSize = outputShape[0]; in BatchToSpaceNd() local
H A DSpaceToBatchNd.cpp51 const unsigned int outputBatchSize = outputShape[0]; in SpaceToBatchNd() local
/aosp_15_r20/external/armnn/src/armnn/layers/
H A DDepthwiseConvolution2dLayer.cpp96 unsigned int outputBatchSize = inputBatchSize; in InferOutputShapes() local
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DConstantTestImpl.cpp41 constexpr unsigned int outputBatchSize = inputBatchSize; in ConstantTestImpl() local
H A DActivationTestImpl.cpp44 unsigned int outputBatchSize = inputBatchSize; in BoundedReLuTestCommon() local
231 constexpr static unsigned int outputBatchSize = inputBatchSize; member
431 constexpr static unsigned int outputBatchSize = inputBatchSize; in SimpleActivationTest() local
H A DPooling3dTestImpl.cpp72 unsigned int outputBatchSize = armnn::numeric_cast<unsigned int>(outputShape[0]); in SimplePooling3dTestImpl() local
H A DPooling2dTestImpl.cpp59 unsigned int outputBatchSize = armnn::numeric_cast<unsigned int>(outputShape[0]); in SimplePooling2dTestImpl() local
H A DConv2dTestImpl.cpp2044 unsigned int outputBatchSize = inputBatchSize; in DepthwiseConvolution2dTestImpl() local
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/
H A DDepthwiseConvolution2dEndToEndTests.hpp72 unsigned int outputBatchSize = inputBatchSize; in DepthwiseConvolution2dEndToEnd() local
/aosp_15_r20/external/armnn/src/backends/backendsCommon/
H A DWorkloadData.cpp1624 const unsigned int outputBatchSize = outputTensorInfo.GetShape()[0]; in Validate() local