Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DQuantizedLSTMTest.cpp60 const uint32_t numBatches = inputOperandTypeParams[0].shape[0]; in QuantizedLSTMOpModel() local
240 const int numBatches = input.size(); in VerifyGoldens() local
272 const int numBatches = 2; in TEST_F() local
DRoiAlign.cpp58 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in roiAlignNhwc() local
185 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in roiAlignQuantNhwc() local
340 uint32_t numBatches = getSizeOfDimension(input, 0); in prepare() local
DRoiPooling.cpp52 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in roiPoolingNhwc() local
186 [[maybe_unused]] uint32_t numBatches = getSizeOfDimension(input, 0); in prepare() local
DInstanceNormalization.cpp42 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in instanceNormNhwc() local
DGenerateProposals.cpp72 uint32_t numBatches = getSizeOfDimension(imageInfoDataShape, 0); in bboxTransformFloat32() local
214 uint32_t numBatches = getSizeOfDimension(imageInfoShape, 0); in prepare() local
856 uint32_t numBatches = getSizeOfDimension(scoresShape, 0); in generateProposalsNhwcFloat32Compute() local
1129 uint32_t numBatches = getSizeOfDimension(scoreShape, 0); in prepare() local
1263 uint32_t numBatches = getSizeOfDimension(scoreShape, 0); in detectionPostprocessFloat32() local
1403 uint32_t numBatches = getSizeOfDimension(scoreShape, 0); in prepare() local
DConv2D.cpp359 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in convQuant8PerChannelNhwc() local
452 [[maybe_unused]] uint32_t numBatches = getSizeOfDimension(inputShape, 0); in convQuant8PerChannelNhwc() local
DQuantizedLSTM.cpp255 const uint32_t numBatches = SizeOfDimension(input, 0); in prepare() local
DDepthwiseConv2D.cpp285 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in depthwiseConvQuant8PerChannelNhwc() local
/packages/modules/NeuralNetworks/runtime/test/
DTestValidateOperations.cpp3529 const int numBatches = 2; in detectionPostprocessingOpTest() local