Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DQuantizedLSTMTest.cpp59 const uint32_t numBatches = inputOperandTypeParams[0].shape[0]; in QuantizedLSTMOpModel() local
239 const int numBatches = input.size(); in VerifyGoldens() local
271 const int numBatches = 2; in TEST_F() local
DRoiPooling.cpp64 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in roiPoolingNhwc() local
237 uint32_t numBatches = getSizeOfDimension(input, 0); in prepare() local
DRoiAlign.cpp68 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in roiAlignNhwc() local
195 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in roiAlignQuantNhwc() local
392 uint32_t numBatches = getSizeOfDimension(input, 0); in prepare() local
DInstanceNormalization.cpp51 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in instanceNormNhwc() local
DGenerateProposals.cpp68 uint32_t numBatches = getSizeOfDimension(imageInfoDataShape, 0); in bboxTransformFloat32() local
245 uint32_t numBatches = getSizeOfDimension(imageInfoShape, 0); in prepare() local
962 uint32_t numBatches = getSizeOfDimension(scoresShape, 0); in generateProposalsNhwcFloat32Compute() local
1296 uint32_t numBatches = getSizeOfDimension(scoreShape, 0); in prepare() local
1449 uint32_t numBatches = getSizeOfDimension(scoreShape, 0); in detectionPostprocessFloat32() local
1620 uint32_t numBatches = getSizeOfDimension(scoreShape, 0); in prepare() local
DQuantizedLSTM.cpp254 const uint32_t numBatches = SizeOfDimension(input, 0); in prepare() local
DConv2D.cpp361 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in convQuant8PerChannelNhwc() local
454 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in convQuant8PerChannelNhwc() local
DDepthwiseConv2D.cpp287 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in depthwiseConvQuant8PerChannelNhwc() local
/frameworks/base/services/core/java/com/android/server/
DAlarmManagerService.java1028 final int numBatches = batches.size(); in haveBatchesTimeTickAlarm() local
3757 final int numBatches = batches.size(); in recordWakeupAlarms() local
/frameworks/ml/nn/runtime/test/
DTestValidateOperations.cpp3491 const int numBatches = 2; in detectionPostprocessingOpTest() local