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.cpp58 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in roiPoolingNhwc() local
210 uint32_t numBatches = getSizeOfDimension(input, 0); in prepare() local
DRoiAlign.cpp62 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in roiAlignNhwc() local
190 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in roiAlignNhwc() local
375 uint32_t numBatches = getSizeOfDimension(input, 0); in prepare() local
DInstanceNormalization.cpp49 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in instanceNormNhwc() local
DGenerateProposals.cpp61 uint32_t numBatches = getSizeOfDimension(imageInfoDataShape, 0); in bboxTransformFloat32() local
209 uint32_t numBatches = getSizeOfDimension(imageInfoShape, 0); in prepare() local
853 uint32_t numBatches = getSizeOfDimension(scoresShape, 0); in generateProposalsNhwcFloat32Compute() local
1180 uint32_t numBatches = getSizeOfDimension(scoreShape, 0); in prepare() local
1316 uint32_t numBatches = getSizeOfDimension(scoreShape, 0); in detectionPostprocessFloat32() local
1487 uint32_t numBatches = getSizeOfDimension(scoreShape, 0); in prepare() local
DDepthwiseConv2D.cpp161 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in depthwiseConvQuant8PerChannel() local
DQuantizedLSTM.cpp249 const uint32_t numBatches = SizeOfDimension(input, 0); in prepare() local
DConv2D.cpp292 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in convQuant8PerChannelNhwc() local
/frameworks/base/services/core/java/com/android/server/
DAlarmManagerService.java1029 final int numBatches = batches.size(); in haveBatchesTimeTickAlarm() local
3760 final int numBatches = batches.size(); in recordWakeupAlarms() local
/frameworks/ml/nn/runtime/test/
DTestValidateOperations.cpp2523 const int numBatches = 2; in detectionPostprocessingOpTest() local