Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/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
DRoiPooling.cpp65 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in roiPoolingNhwc() local
238 uint32_t numBatches = getSizeOfDimension(input, 0); in prepare() local
DRoiAlign.cpp70 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in roiAlignNhwc() local
197 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in roiAlignQuantNhwc() local
395 uint32_t numBatches = getSizeOfDimension(input, 0); in prepare() local
DInstanceNormalization.cpp52 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in instanceNormNhwc() local
DGenerateProposals.cpp69 uint32_t numBatches = getSizeOfDimension(imageInfoDataShape, 0); in bboxTransformFloat32() local
246 uint32_t numBatches = getSizeOfDimension(imageInfoShape, 0); in prepare() local
969 uint32_t numBatches = getSizeOfDimension(scoresShape, 0); in generateProposalsNhwcFloat32Compute() local
1305 uint32_t numBatches = getSizeOfDimension(scoreShape, 0); in prepare() local
1460 uint32_t numBatches = getSizeOfDimension(scoreShape, 0); in detectionPostprocessFloat32() local
1633 uint32_t numBatches = getSizeOfDimension(scoreShape, 0); in prepare() local
DQuantizedLSTM.cpp251 const uint32_t numBatches = SizeOfDimension(input, 0); in prepare() local
DConv2D.cpp362 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in convQuant8PerChannelNhwc() local
455 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in convQuant8PerChannelNhwc() local
DDepthwiseConv2D.cpp289 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in depthwiseConvQuant8PerChannelNhwc() local
/packages/modules/NeuralNetworks/runtime/test/
DTestValidateOperations.cpp3490 const int numBatches = 2; in detectionPostprocessingOpTest() local