| /frameworks/ml/nn/common/operations/ |
| D | QuantizedLSTMTest.cpp | 59 const uint32_t numBatches = inputOperandTypeParams[0].shape[0]; in QuantizedLSTMOpModel() local
66 OperandType cellStateOutOperandType(Type::TENSOR_QUANT16_SYMM, {numBatches, outputSize}, in QuantizedLSTMOpModel()
69 OperandType outputOperandType(Type::TENSOR_QUANT8_ASYMM, {numBatches, outputSize}, in QuantizedLSTMOpModel()
82 cellStateOut_.resize(numBatches * outputSize, 0); in QuantizedLSTMOpModel()
83 output_.resize(numBatches * outputSize, 0); in QuantizedLSTMOpModel()
239 const int numBatches = input.size(); in VerifyGoldens() local
240 EXPECT_GT(numBatches, 0); in VerifyGoldens()
247 for (int b = 0; b < numBatches; ++b) { in VerifyGoldens()
257 for (int b = 0; b < numBatches; ++b) { in VerifyGoldens()
271 const int numBatches = 2; in TEST_F() local
[all …]
|
| D | RoiAlign.cpp | 62 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in roiAlignNhwc() local
81 NN_RET_CHECK_LT(batchId, numBatches); in roiAlignNhwc()
190 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in roiAlignNhwc() local
214 NN_RET_CHECK_LT(batchId, numBatches); in roiAlignNhwc()
375 uint32_t numBatches = getSizeOfDimension(input, 0); in prepare() local
381 NN_RET_CHECK_GT(numBatches, 0); in prepare()
|
| D | GroupedConv2D.cpp | 30 uint32_t numBatches = getSizeOfDimension(inputShape, 0); \
56 for (uint32_t b = 0; b < numBatches; b++) { in groupedConvFloat32()
127 for (uint32_t b = 0; b < numBatches; b++) { in groupedConvQuant8()
213 for (uint32_t b = 0; b < numBatches; b++) { in groupedConvQuant8PerChannel()
|
| D | TransposeConv2D.cpp | 101 uint32_t numBatches = getSizeOfDimension(inputShape, 0); \
129 for (uint32_t b = 0; b < numBatches; b++) { in transposeConvNhwc()
159 const uint32_t outerSize = numBatches * outputHeight * outputWidth; in transposeConvNhwc()
215 for (uint32_t b = 0; b < numBatches; b++) { in transposeConvNhwc()
250 const uint32_t outerSize = numBatches * outputHeight * outputWidth; in transposeConvNhwc()
356 for (uint32_t b = 0; b < numBatches; b++) { in transposeConvQuant8PerChannelNhwc()
390 const uint32_t outerSize = numBatches * outputHeight * outputWidth; in transposeConvQuant8PerChannelNhwc()
|
| D | InstanceNormalization.cpp | 49 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in instanceNormNhwc() local
53 for (uint32_t b = 0; b < numBatches; b++) { in instanceNormNhwc()
|
| D | GenerateProposals.cpp | 61 uint32_t numBatches = getSizeOfDimension(imageInfoDataShape, 0); in bboxTransformFloat32() local
73 NN_RET_CHECK_LT(batchIndex, numBatches); in bboxTransformFloat32()
209 uint32_t numBatches = getSizeOfDimension(imageInfoShape, 0); in prepare() local
211 NN_RET_CHECK_GT(numBatches, 0); in prepare()
853 uint32_t numBatches = getSizeOfDimension(scoresShape, 0); in generateProposalsNhwcFloat32Compute() local
896 for (uint32_t b = 0; b < numBatches; b++) { in generateProposalsNhwcFloat32Compute()
1180 uint32_t numBatches = getSizeOfDimension(scoreShape, 0); in prepare() local
1185 NN_RET_CHECK_EQ(getSizeOfDimension(bboxDeltasShape, 0), numBatches); in prepare()
1189 NN_RET_CHECK_EQ(getSizeOfDimension(imageInfoDataShape, 0), numBatches); in prepare()
1316 uint32_t numBatches = getSizeOfDimension(scoreShape, 0); in detectionPostprocessFloat32() local
[all …]
|
| D | RoiPooling.cpp | 58 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in roiPoolingNhwc() local
77 NN_RET_CHECK_LT(batchId, numBatches); in roiPoolingNhwc()
210 uint32_t numBatches = getSizeOfDimension(input, 0); in prepare() local
|
| D | QuantizedLSTM.cpp | 249 const uint32_t numBatches = SizeOfDimension(input, 0); in prepare() local
254 NN_RET_CHECK_EQ(SizeOfDimension(prevOutput, 0), numBatches); in prepare()
314 NN_CHECK_EQ(SizeOfDimension(prevCellState, 0), numBatches); in prepare()
|
| D | DepthwiseConv2D.cpp | 161 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in depthwiseConvQuant8PerChannel() local
197 for (uint32_t b = 0; b < numBatches; b++) { in depthwiseConvQuant8PerChannel()
|
| D | Conv2D.cpp | 292 uint32_t numBatches = getSizeOfDimension(inputShape, 0); in convQuant8PerChannelNhwc() local
327 for (uint32_t b = 0; b < numBatches; b++) { in convQuant8PerChannelNhwc()
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 2523 const int numBatches = 2; in detectionPostprocessingOpTest() local
2528 uint32_t inputDims[3] = {numBatches, numAnchors, numClasses}; in detectionPostprocessingOpTest()
2530 uint32_t deltasDims[3] = {numBatches, numAnchors, lengthBoxEncoding}; in detectionPostprocessingOpTest()
2547 uint32_t outputScoreDims[2] = {numBatches, maxNumDetectionsValue}; in detectionPostprocessingOpTest()
2549 uint32_t boundingBoxesDims[3] = {numBatches, maxNumDetectionsValue, 4}; in detectionPostprocessingOpTest()
2553 uint32_t numValidDims[1] = {numBatches}; in detectionPostprocessingOpTest()
|
| /frameworks/base/services/core/java/com/android/server/ |
| D | AlarmManagerService.java | 1029 final int numBatches = batches.size(); in haveBatchesTimeTickAlarm() local
1030 for (int i = 0; i < numBatches; i++) { in haveBatchesTimeTickAlarm()
3760 final int numBatches = batches.size(); in recordWakeupAlarms() local
3761 for (int nextBatch = 0; nextBatch < numBatches; nextBatch++) { in recordWakeupAlarms()
|