Home
last modified time | relevance | path

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

/external/armnn/src/backends/neon/test/
DNeonTimerTest.cpp58 unsigned int outputBatchSize = inputBatchSize; in BOOST_AUTO_TEST_CASE() local
/external/armnn/src/backends/reference/workloads/
DBatchToSpaceNd.cpp54 const unsigned int outputBatchSize = outputShape[0]; in BatchToSpaceNd() local
DSpaceToBatchNd.cpp51 const unsigned int outputBatchSize = outputShape[0]; in SpaceToBatchNd() local
/external/armnn/src/backends/backendsCommon/test/layerTests/
DConstantTestImpl.cpp41 constexpr unsigned int outputBatchSize = inputBatchSize; in ConstantTestImpl() local
DActivationTestImpl.cpp46 unsigned int outputBatchSize = inputBatchSize; in BoundedReLuTestCommon() local
233 constexpr static unsigned int outputBatchSize = inputBatchSize; member
430 constexpr static unsigned int outputBatchSize = inputBatchSize; in SimpleActivationTest() local
DPooling2dTestImpl.cpp58 unsigned int outputBatchSize = armnn::numeric_cast<unsigned int>(outputExpected.shape()[0]); in SimplePooling2dTestImpl() local
DConv2dTestImpl.cpp1943 unsigned int outputBatchSize = inputBatchSize; in DepthwiseConvolution2dTestImpl() local
/external/armnn/src/armnn/layers/
DDepthwiseConvolution2dLayer.cpp116 unsigned int outputBatchSize = inputBatchSize; in InferOutputShapes() local
/external/armnn/src/backends/backendsCommon/
DWorkloadData.cpp1455 const unsigned int outputBatchSize = outputTensorInfo.GetShape()[0]; in Validate() local
1502 const unsigned int outputBatchSize = outputTensorInfo.GetShape()[0]; in Validate() local