Home
last modified time | relevance | path

Searched refs:batchSize (Results 1 – 25 of 28) sorted by relevance

12

/packages/modules/NeuralNetworks/common/cpu_operations/
DQLSTM.cpp67 const uint32_t batchSize = getSizeOfDimension(inputShape, 0); in prepare() local
184 NN_RET_CHECK_EQ(getSizeOfDimension(outputStateShape, 0), batchSize); in prepare()
188 NN_RET_CHECK_EQ(getSizeOfDimension(cellStateShape, 0), batchSize); in prepare()
276 const uint32_t batchSize = inputShape.dimensions[0]; in execute() local
541 std::vector<int16_t> inputGateBuffer(batchSize * numUnits); in execute()
542 std::vector<int16_t> forgetGateBuffer(batchSize * numUnits); in execute()
543 std::vector<int16_t> cellGateBuffer(batchSize * numUnits); in execute()
544 std::vector<int16_t> outputGateBuffer(batchSize * numUnits); in execute()
545 std::vector<int8_t> buffer8(batchSize * numUnits); in execute()
560 inputToForgetEffectiveScaleB, batchSize, inputSize, in execute()
[all …]
DBidirectionalSequenceRNN.cpp189 const uint32_t batchSize = getSizeOfDimension(inputShape, 1); in executeTyped() local
221 tempHiddenState.resize(std::max(batchSize * fwNumUnits, batchSize * bwNumUnits)); in executeTyped()
228 const T* inputBatchPtr = input + i * batchSize * inputSize; in executeTyped()
231 auxInputBatchPtr = auxInput + i * batchSize * auxInputSize; in executeTyped()
234 T* fwOutputBatchPtr = fwOutput + i * batchSize * fwOutputBatchStride; in executeTyped()
247 const T* inputBatchPtr = bwInput + i * batchSize * inputSize; in executeTyped()
250 auxInputBatchPtr = auxInput + i * batchSize * auxInputSize; in executeTyped()
258 bwOutputBatchPtr = fwOutput + i * batchSize * bwOutputBatchStride; in executeTyped()
261 bwOutputBatchPtr = bwOutput + i * batchSize * bwOutputBatchStride; in executeTyped()
326 const uint32_t batchSize = in prepare() local
[all …]
DUnidirectionalSequenceRNN.cpp85 const uint32_t batchSize = getSizeOfDimension(inputShape, 1); in executeTyped() local
98 input += batchSize * inputSize; in executeTyped()
100 output += batchSize * numUnits; in executeTyped()
111 std::copy(hiddenState, hiddenState + batchSize * numUnits, stateOutput); in executeTyped()
127 const uint32_t batchSize = in prepare() local
144 NN_RET_CHECK_EQ(batchSize, getSizeOfDimension(hiddenState, 0)); in prepare()
149 output.dimensions[0] = timeMajor ? maxTime : batchSize; in prepare()
150 output.dimensions[1] = timeMajor ? batchSize : maxTime; in prepare()
157 outputStateShape.dimensions[0] = batchSize; in prepare()
DLSTM.cpp434 const uint32_t batchSize = (inputRank == 3) ? getSizeOfDimension(input_shape, timeMajor ? 1 : 0) in LSTMEvalFloat32() local
441 batchInputShape.dimensions = {batchSize, inputSize}; in LSTMEvalFloat32()
442 const uint32_t batchInputSize = batchSize * inputSize; in LSTMEvalFloat32()
443 const uint32_t batchOutputSize = batchSize * outputSize; in LSTMEvalFloat32()
470 output_state_in_buffer, output_state_in_buffer + batchSize * outputSize); in LSTMEvalFloat32()
472 cell_state_in_buffer + batchSize * numCells); in LSTMEvalFloat32()
507 output_state_out_buffer + batchSize * outputSize); in LSTMEvalFloat32()
509 cell_state_out_buffer + batchSize * numCells); in LSTMEvalFloat32()
554 const uint32_t batchSize = (inputRank == 3) ? getSizeOfDimension(input_shape, timeMajor ? 1 : 0) in LSTMEvalFloat16() local
561 batchInputShape.dimensions = {batchSize, inputSize}; in LSTMEvalFloat16()
[all …]
DUnidirectionalSequenceLSTM.cpp96 const uint32_t batchSize = getSizeOfDimension(inputShape, isTimeMajor(context) ? 1 : 0); in prepare() local
213 NN_RET_CHECK_EQ(getSizeOfDimension(outputStateShape, 0), batchSize); in prepare()
217 NN_RET_CHECK_EQ(getSizeOfDimension(cellStateShape, 0), batchSize); in prepare()
277 outputStateOutTensor.dimensions[0] = batchSize; in prepare()
283 cellStateOutTensor.dimensions[0] = batchSize; in prepare()
DResizeImageOps.cpp63 const int batchSize = getSizeOfDimension(inputShape, 0); in resizeNearestNeighbor() local
76 for (int b = 0; b < batchSize; ++b) { in resizeNearestNeighbor()
DGenerateProposals.cpp862 uint32_t batchSize = height * width * numAnchors; in generateProposalsNhwcFloat32Compute() local
863 uint32_t roiBufferSize = batchSize * kRoiDim; in generateProposalsNhwcFloat32Compute()
891 tempRoiShape.dimensions = {batchSize, kRoiDim}; in generateProposalsNhwcFloat32Compute()
893 tempBBoxDeltasShape.dimensions = {batchSize, kRoiDim}; in generateProposalsNhwcFloat32Compute()
894 std::vector<int32_t> tempBatchSplitData(batchSize, 0); in generateProposalsNhwcFloat32Compute()
895 Shape tempbatchSplitShape = {.dimensions = {batchSize}}; in generateProposalsNhwcFloat32Compute()
911 std::vector<uint32_t> select(batchSize); in generateProposalsNhwcFloat32Compute()
941 scoresBase += batchSize; in generateProposalsNhwcFloat32Compute()
/packages/modules/AppSearch/testing/contactsindexertests/src/com/android/server/appsearch/contactsindexer/
DContactsIndexerImplTest.java119 int batchSize = 5; in testBatcher_noFlushBeforeReachingLimit() local
120 ContactsBatcher batcher = new ContactsBatcher(mAppSearchHelper, batchSize, in testBatcher_noFlushBeforeReachingLimit()
123 for (int i = 0; i < batchSize - 1; ++i) { in testBatcher_noFlushBeforeReachingLimit()
136 int batchSize = 5; in testBatcher_autoFlush() local
137 ContactsBatcher batcher = new ContactsBatcher(mAppSearchHelper, batchSize, in testBatcher_autoFlush()
140 for (int i = 0; i < batchSize; ++i) { in testBatcher_autoFlush()
150 assertThat(mAppSearchHelper.mIndexedContacts).hasSize(batchSize); in testBatcher_autoFlush()
157 int batchSize = 2; in testBatcher_contactFingerprintSame_notIndexed() local
158 ContactsBatcher batcher = new ContactsBatcher(mAppSearchHelper, batchSize, in testBatcher_contactFingerprintSame_notIndexed()
183 int batchSize = 2; in testBatcher_contactFingerprintDifferent_notIndexedButBatched() local
[all …]
/packages/modules/RemoteKeyProvisioning/app/src/com/android/rkpdapp/interfaces/
DSystemInterface.java183 int batchSize = mBinder.getHardwareInfo().supportedNumKeysInCsr; in getBatchSize() local
185 if (batchSize < RpcHardwareInfo.MIN_SUPPORTED_NUM_KEYS_IN_CSR) { in getBatchSize()
186 Log.w(TAG, "HAL returned a batch size that's too small (" + batchSize in getBatchSize()
191 if (batchSize > maxBatchSize) { in getBatchSize()
192 Log.w(TAG, "HAL returned a batch size that's too large (" + batchSize in getBatchSize()
197 return batchSize; in getBatchSize()
/packages/services/Car/cpp/telemetry/cartelemetryd/sampleclient/
Dmain.cpp74 int batchSize = 0; in main() local
82 argError = sscanf(optarg, "%d", &batchSize) != 1; in main()
104 if (batchSize == 0) { in main()
156 for (int i = 0; i < batchSize; i++) { in main()
176 << batchSize << " carData, each with payload of " << cardataSize << " bytes" in main()
/packages/modules/OnDevicePersonalization/framework/java/android/adservices/ondevicepersonalization/
DInferenceInputParcel.java125 int batchSize, in InferenceInputParcel() argument
136 this.mBatchSize = batchSize; in InferenceInputParcel()
235 int batchSize = in.readInt(); in InferenceInputParcel() local
248 this.mBatchSize = batchSize; in InferenceInputParcel()
DInferenceInput.java354 int batchSize, in InferenceInput() argument
358 this.mBatchSize = batchSize; in InferenceInput()
/packages/apps/QuickSearchBox/src/com/android/quicksearchbox/util/
DBatchingNamedTaskExecutor.kt45 fun executeNextBatch(batchSize: Int) { in executeNextBatch()
48 val count: Int = Math.min(mQueuedTasks.size, batchSize) in executeNextBatch()
/packages/apps/Calendar/src/com/android/calendar/alerts/
DAlarmScheduler.kt108 batchSize: Int, in scheduleNextAlarm()
123 batchSize, in scheduleNextAlarm()
190 batchSize: Int, in queryNextReminderAndSchedule()
214 while (index++ < batchSize && instancesCursor.moveToNext()) { in queryNextReminderAndSchedule()
/packages/modules/RemoteKeyProvisioning/app/src/com/android/rkpdapp/provisioner/
DProvisioner.java139 int batchSize = Math.min(keysGenerated.size() - provisionedSoFar, maxBatchSize); in fetchCertificates() local
141 keysGenerated.subList(provisionedSoFar, batchSize + provisionedSoFar))); in fetchCertificates()
142 provisionedSoFar += batchSize; in fetchCertificates()
/packages/modules/AdServices/adservices/service-core/java/com/android/adservices/service/stats/kanon/
DKAnonSignStatusStats.java55 public abstract Builder setBatchSize(int batchSize); in setBatchSize() argument
/packages/modules/OnDevicePersonalization/src/com/android/ondevicepersonalization/services/inference/
DIsolatedModelServiceImpl.java153 int batchSize = inputParcel.getBatchSize(); in runTfliteInterpreter() local
157 shape[0] = batchSize; in runTfliteInterpreter()
/packages/modules/RemoteKeyProvisioning/app/tests/unit/src/com/android/rkpdapp/unittest/
DProvisionerTest.java104 final int batchSize = 13; in testProvisionerUsesCorrectBatchSize() local
107 doReturn(batchSize).when(mockSystem).getBatchSize(); in testProvisionerUsesCorrectBatchSize()
/packages/modules/Bluetooth/android/app/src/com/android/bluetooth/opp/
DBluetoothOppReceiver.java91 int batchSize = mOppManager.getBatchSize(); in onReceive() local
95 R.string.bt_toast_5, Integer.toString(batchSize), deviceName); in onReceive()
/packages/services/DeviceAsWebcam/interface/jni/
DEncoder.cpp246 const uint32_t batchSize = DCTSIZE * maxVSampFactor; in i420ToJpeg() local
252 jpeg_write_raw_data(cInfo.get(), planes, batchSize); in i420ToJpeg()
/packages/apps/Contacts/src/com/android/contacts/
DContactSaveService.java1286 final int batchSize = MAX_CONTACTS_PROVIDER_BATCH_SIZE; in splitContact() local
1288 final ArrayList<ContentProviderOperation> operations = new ArrayList<>(batchSize); in splitContact()
1331 final int batchSize = MAX_CONTACTS_PROVIDER_BATCH_SIZE; in buildSplitTwoContacts() local
1336 if (operations.size() > 0 && operations.size() % batchSize == 0) { in buildSplitTwoContacts()
1434 final int batchSize = MAX_CONTACTS_PROVIDER_BATCH_SIZE; in joinSeveralContacts() local
1435 final ArrayList<ContentProviderOperation> operations = new ArrayList<>(batchSize); in joinSeveralContacts()
1442 if (operations.size() > 0 && operations.size() % batchSize == 0) { in joinSeveralContacts()
/packages/modules/AppSearch/service/java/com/android/server/appsearch/contactsindexer/
DContactsIndexerImpl.java426 int batchSize, in ContactsBatcher() argument
429 mBatchSize = batchSize; in ContactsBatcher()
/packages/modules/NeuralNetworks/runtime/test/
DTestValidateOperations.cpp3964 const uint32_t batchSize = 2; in bidirectionlSequenceRNNTest() local
3969 uint32_t inputDims[3] = {maxTime, batchSize, inputSize}; in bidirectionlSequenceRNNTest()
3973 uint32_t hiddenStateDims[2] = {batchSize, numUnits}; in bidirectionlSequenceRNNTest()
3974 uint32_t outputDims[2] = {batchSize, numUnits}; in bidirectionlSequenceRNNTest()
4040 const uint32_t batchSize = 2; in unidirectionlSequenceRNNTest() local
4045 uint32_t inputDims[3] = {maxTime, batchSize, inputSize}; in unidirectionlSequenceRNNTest()
4049 uint32_t hiddenStateDims[2] = {batchSize, numUnits}; in unidirectionlSequenceRNNTest()
4050 uint32_t outputDims[2] = {batchSize, numUnits}; in unidirectionlSequenceRNNTest()
4106 const uint32_t batchSize = 3; in unidirectionalSequenceLSTMTest() local
4111 uint32_t inputDims[3] = {maxTime, batchSize, inputSize}; in unidirectionalSequenceLSTMTest()
[all …]
/packages/apps/DocumentsUI/src/com/android/documentsui/services/
DCopyJob.java322 final boolean verifySpaceAvailable(long batchSize) { in verifySpaceAvailable() argument
326 if (batchSize >= 0) { in verifySpaceAvailable()
334 available = (batchSize <= root.availableBytes); in verifySpaceAvailable()
/packages/apps/TV/src/com/android/tv/data/epg/
DEpgFetcherImpl.java458 int batchSize = (int) Math.max(1, mBackendKnobsFlags.epgFetcherChannelsPerProgramFetch()); in batchFetchEpg() local
459 for (Iterable<EpgChannel> batch : Iterables.partition(epgChannels, batchSize)) { in batchFetchEpg()

12