Home
last modified time | relevance | path

Searched defs:inputBatchSize (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/layerTests/
H A DActivationTestImpl.cpp38 unsigned int inputBatchSize) in BoundedReLuTestCommon()
100 unsigned int inputBatchSize = 1; in BoundedReLuUpperAndLowerBoundTest() local
132 unsigned int inputBatchSize = 1; in BoundedReLuUpperBoundOnlyTest() local
164 unsigned int inputBatchSize = 1; in BoundedReLuUint8UpperBoundOnlyTest() local
196 unsigned int inputBatchSize = 1; in BoundedReLuUint8UpperAndLowerBoundTest() local
226 constexpr static unsigned int inputBatchSize = 2; member
426 constexpr static unsigned int inputBatchSize = 1u; in SimpleActivationTest() local
H A DConstantTestImpl.cpp36 constexpr unsigned int inputBatchSize = 2; in ConstantTestImpl() local
H A DPooling3dTestImpl.cpp66 unsigned int inputBatchSize = armnn::numeric_cast<unsigned int>(inputShape[0]); in SimplePooling3dTestImpl() local
H A DPooling2dTestImpl.cpp54 unsigned int inputBatchSize = armnn::numeric_cast<unsigned int>(inputShape[0]); in SimplePooling2dTestImpl() local
H A DConv2dTestImpl.cpp2036 unsigned int inputBatchSize = 1; in DepthwiseConvolution2dTestImpl() local
/aosp_15_r20/external/armnn/src/backends/neon/test/
H A DNeonTimerTest.cpp45 unsigned int inputBatchSize = 1u; variable
/aosp_15_r20/external/armnn/src/backends/reference/workloads/
H A DSpaceToDepth.cpp47 const unsigned int inputBatchSize = inputShape[0]; in SpaceToDepth() local
H A DBatchToSpaceNd.cpp51 const unsigned int inputBatchSize = inputShape[0]; in BatchToSpaceNd() local
H A DSpaceToBatchNd.cpp47 const unsigned int inputBatchSize = inputShape[0]; in SpaceToBatchNd() local
/aosp_15_r20/external/armnn/src/armnn/layers/
H A DDepthwiseConvolution2dLayer.cpp78 unsigned int inputBatchSize = inputShape[0]; in InferOutputShapes() local
/aosp_15_r20/external/armnn/src/backends/backendsCommon/test/
H A DDepthwiseConvolution2dEndToEndTests.hpp64 unsigned int inputBatchSize = 1; in DepthwiseConvolution2dEndToEnd() local
/aosp_15_r20/external/armnn/src/backends/backendsCommon/
H A DWorkloadData.cpp1623 const unsigned int inputBatchSize = inputTensorInfo.GetShape()[0]; in Validate() local