Home
last modified time | relevance | path

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

123

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DQLSTM.cpp183 const uint32_t batchSize = getSizeOfDimension(inputShape, 0); in prepare() local
300 NN_RET_CHECK_EQ(getSizeOfDimension(outputStateShape, 0), batchSize); in prepare()
304 NN_RET_CHECK_EQ(getSizeOfDimension(cellStateShape, 0), batchSize); in prepare()
392 const uint32_t batchSize = inputShape.dimensions[0]; in execute() local
657 std::vector<int16_t> inputGateBuffer(batchSize * numUnits); in execute()
658 std::vector<int16_t> forgetGateBuffer(batchSize * numUnits); in execute()
659 std::vector<int16_t> cellGateBuffer(batchSize * numUnits); in execute()
660 std::vector<int16_t> outputGateBuffer(batchSize * numUnits); in execute()
661 std::vector<int8_t> buffer8(batchSize * numUnits); in execute()
735 batchSize, numUnits, inputGateBuffer.data()); in execute()
[all …]
H A DBidirectionalSequenceRNN.cpp218 const uint32_t batchSize = getSizeOfDimension(inputShape, 1); in executeTyped() local
250 tempHiddenState.resize(std::max(batchSize * fwNumUnits, batchSize * bwNumUnits)); in executeTyped()
257 const T* inputBatchPtr = input + i * batchSize * inputSize; in executeTyped()
276 const T* inputBatchPtr = bwInput + i * batchSize * inputSize; in executeTyped()
385 const uint32_t batchSize = in prepare() local
407 NN_RET_CHECK_EQ(batchSize, getSizeOfDimension(fwHiddenState, 0)); in prepare()
416 NN_RET_CHECK_EQ(batchSize, getSizeOfDimension(bwHiddenState, 0)); in prepare()
440 fwOutput.dimensions[0] = timeMajor ? maxTime : batchSize; in prepare()
441 fwOutput.dimensions[1] = timeMajor ? batchSize : maxTime; in prepare()
447 bwOutput.dimensions[0] = timeMajor ? maxTime : batchSize; in prepare()
[all …]
H A DUnidirectionalSequenceRNN.cpp97 const uint32_t batchSize = getSizeOfDimension(inputShape, 1); in executeTyped() local
110 input += batchSize * inputSize; in executeTyped()
112 output += batchSize * numUnits; in executeTyped()
123 std::copy(hiddenState, hiddenState + batchSize * numUnits, stateOutput); in executeTyped()
162 const uint32_t batchSize = in prepare() local
179 NN_RET_CHECK_EQ(batchSize, getSizeOfDimension(hiddenState, 0)); in prepare()
184 output.dimensions[0] = timeMajor ? maxTime : batchSize; in prepare()
185 output.dimensions[1] = timeMajor ? batchSize : maxTime; in prepare()
192 outputStateShape.dimensions[0] = batchSize; in prepare()
H A DLSTM.cpp441 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()
509 cell_state_out_buffer + batchSize * numCells); in LSTMEvalFloat32()
561 batchInputShape.dimensions = {batchSize, inputSize}; in LSTMEvalFloat16()
562 const uint32_t batchInputSize = batchSize * inputSize; in LSTMEvalFloat16()
563 const uint32_t batchOutputSize = batchSize * outputSize; in LSTMEvalFloat16()
671 std::vector<float> cell_state_out_float32(batchSize * numCells); in LSTMEvalFloat16()
706 std::vector<float> outputStateInCurrentTimeStep(batchSize * outputSize); in LSTMEvalFloat16()
[all …]
H A DUnidirectionalSequenceLSTM.cpp205 const uint32_t batchSize = getSizeOfDimension(inputShape, isTimeMajor(context) ? 1 : 0); in prepare() local
322 NN_RET_CHECK_EQ(getSizeOfDimension(outputStateShape, 0), batchSize); in prepare()
326 NN_RET_CHECK_EQ(getSizeOfDimension(cellStateShape, 0), batchSize); in prepare()
386 outputStateOutTensor.dimensions[0] = batchSize; in prepare()
392 cellStateOutTensor.dimensions[0] = batchSize; in prepare()
/aosp12/frameworks/base/core/java/com/android/internal/os/
H A DBinderCallHeavyHitterWatcher.java247 public void setConfig(final boolean enable, final int batchSize, final float threshold, in setConfig() argument
263 if (batchSize == mTotalInputSize && Math.abs(threshold - mThreshold) < EPSILON) { in setConfig()
272 int inputSize = batchSize; in setConfig()
274 inputSize = (int) (batchSize * (1 - validationRatio)); in setConfig()
277 sketch.setConfig(batchSize, capacity); in setConfig()
281 + batchSize + ", " + capacity); in setConfig()
285 resetInternalLocked(listener, sketch, inputSize, batchSize, threshold, capacity); in setConfig()
292 final int batchSize, final float threshold, final int capacity) { in resetInternalLocked() argument
300 mTotalInputSize = batchSize; in resetInternalLocked()
/aosp12/hardware/qcom/sm8150p/gps/batching/
H A DBatchingAdapter.h141 void setBatchSize(size_t batchSize) { mBatchSize = batchSize; } in setBatchSize() argument
143 void setTripBatchSize(size_t batchSize) { mTripBatchSize = batchSize; } in setTripBatchSize() argument
/aosp12/hardware/qcom/sm7250/gps/batching/
H A DBatchingAdapter.h141 void setBatchSize(size_t batchSize) { mBatchSize = batchSize; } in setBatchSize() argument
143 void setTripBatchSize(size_t batchSize) { mTripBatchSize = batchSize; } in setTripBatchSize() argument
/aosp12/hardware/qcom/sm7150/gps/batching/
H A DBatchingAdapter.h141 void setBatchSize(size_t batchSize) { mBatchSize = batchSize; } in setBatchSize() argument
143 void setTripBatchSize(size_t batchSize) { mTripBatchSize = batchSize; } in setTripBatchSize() argument
/aosp12/hardware/qcom/sm8150/gps/batching/
H A DBatchingAdapter.h141 void setBatchSize(size_t batchSize) { mBatchSize = batchSize; } in setBatchSize() argument
143 void setTripBatchSize(size_t batchSize) { mTripBatchSize = batchSize; } in setTripBatchSize() argument
/aosp12/packages/apps/RemoteProvisioner/src/com/android/remoteprovisioner/
H A DPeriodicProvisioner.java159 int batchSize = min(keysToCertify, SAFE_CSR_BATCH_SIZE); in run() local
160 Log.i(TAG, "Requesting " + batchSize + " keys to be provisioned."); in run()
161 Provisioner.provisionCerts(batchSize, in run()
167 keysToCertify -= batchSize; in run()
/aosp12/packages/apps/QuickSearchBox/src/com/android/quicksearchbox/util/
H A DBatchingNamedTaskExecutor.java64 public void executeNextBatch(int batchSize) { in executeNextBatch() argument
67 int count = Math.min(mQueuedTasks.size(), batchSize); in executeNextBatch()
/aosp12/frameworks/av/services/camera/libcameraservice/device3/
H A DCamera3OutputStream.cpp668 size_t batchSize = mBatchSize.load(); in getBufferLockedCommon() local
669 if (batchSize == 1) { in getBufferLockedCommon()
680 if (batchSize > remainingBuffers) { in getBufferLockedCommon()
681 batchSize = remainingBuffers; in getBufferLockedCommon()
686 batchedBuffers.resize(batchSize); in getBufferLockedCommon()
1138 status_t Camera3OutputStream::setBatchSize(size_t batchSize) { in setBatchSize() argument
1140 if (batchSize == 0) { in setBatchSize()
1155 if (camera_stream::max_buffers < batchSize) { in setBatchSize()
1158 batchSize = camera_stream::max_buffers; in setBatchSize()
1162 if (!mBatchSize.compare_exchange_strong(defaultBatchSize, batchSize)) { in setBatchSize()
[all …]
H A DCamera3OutputStreamInterface.h110 virtual status_t setBatchSize(size_t batchSize = 1) = 0;
H A DCamera3FakeStream.h101 virtual status_t setBatchSize(size_t batchSize) override;
/aosp12/packages/apps/Calendar/src/com/android/calendar/alerts/
H A DAlarmScheduler.java111 int batchSize, long currentMillis) { in scheduleNextAlarm() argument
118 context.getContentResolver(), alarmManager, batchSize, currentMillis); in scheduleNextAlarm()
178 int batchSize, long currentMillis) { in queryNextReminderAndSchedule() argument
200 while (index++ < batchSize && instancesCursor.moveToNext()) { in queryNextReminderAndSchedule()
/aosp12/hardware/qcom/sdm845/gps/sdm845/android/location_api/
H A DBatchingAPIClient.cpp94 int batchSize = locAPIGetBatchSize(); in getBatchSize() local
95 LOC_LOGd("batchSize: %d", batchSize); in getBatchSize()
96 return batchSize; in getBatchSize()
/aosp12/frameworks/native/services/inputflinger/reader/
H A DInputReader.cpp153 size_t batchSize = 1; in processEventsLocked() local
156 while (batchSize < count) { in processEventsLocked()
157 if (rawEvent[batchSize].type >= EventHubInterface::FIRST_SYNTHETIC_EVENT || in processEventsLocked()
158 rawEvent[batchSize].deviceId != deviceId) { in processEventsLocked()
161 batchSize += 1; in processEventsLocked()
164 ALOGD("BatchSize: %zu Count: %zu", batchSize, count); in processEventsLocked()
166 processEventsForDeviceLocked(deviceId, rawEvent, batchSize); in processEventsLocked()
183 count -= batchSize; in processEventsLocked()
184 rawEvent += batchSize; in processEventsLocked()
/aosp12/hardware/qcom/sm7250/gps/android/2.1/location_api/
H A DBatchingAPIClient.cpp86 int batchSize = locAPIGetBatchSize(); in getBatchSize() local
87 LOC_LOGd("batchSize: %d", batchSize); in getBatchSize()
88 return batchSize; in getBatchSize()
/aosp12/hardware/qcom/sm8150/gps/android/2.0/location_api/
H A DBatchingAPIClient.cpp87 int batchSize = locAPIGetBatchSize(); in getBatchSize() local
88 LOC_LOGd("batchSize: %d", batchSize); in getBatchSize()
89 return batchSize; in getBatchSize()
/aosp12/hardware/qcom/camera/msm8998/QCamera2/HAL3/
H A DQCamera3Stream.cpp379 uint32_t batchSize, in init() argument
388 LOGD("batch size is %d", batchSize); in init()
456 } else if (batchSize) { in init()
457 if (batchSize > MAX_BATCH_SIZE) { in init()
458 LOGE("batchSize:%d is very large", batchSize); in init()
463 mNumBatchBufs = MAX_INFLIGHT_HFR_REQUESTS / batchSize; in init()
465 mStreamInfo->user_buf_info.frame_buf_cnt = batchSize; in init()
472 LOGD("batch size is %d", batchSize); in init()
506 mBatchSize = batchSize; in init()
/aosp12/frameworks/base/packages/SystemUI/src/com/android/systemui/statusbar/notification/collection/notifcollection/
H A DNotifCollectionLogger.kt42 fun logNotifGroupPosted(groupKey: String, batchSize: Int) {
45 int1 = batchSize
/aosp12/hardware/interfaces/camera/device/3.2/default/
H A DCameraDeviceSession.h194 void registerBatch(uint32_t frameNumber, uint32_t batchSize);
215 BufferBatch(uint32_t batchSize) { in BufferBatch()
216 mBuffers.reserve(batchSize); in BufferBatch()
H A DCameraDeviceSession.cpp416 void CameraDeviceSession::ResultBatcher::registerBatch(uint32_t frameNumber, uint32_t batchSize) { in registerBatch() argument
419 batch->mBatchSize = batchSize; in registerBatch()
535 size_t batchSize = 0; in sendBatchBuffersLocked() local
543 if (bb.mBuffers.size() > batchSize) { in sendBatchBuffersLocked()
544 batchSize = bb.mBuffers.size(); in sendBatchBuffersLocked()
552 if (batchSize == 0) { in sendBatchBuffersLocked()
567 results.resize(batchSize); in sendBatchBuffersLocked()
568 for (size_t i = 0; i < batchSize; i++) { in sendBatchBuffersLocked()
/aosp12/hardware/interfaces/gnss/1.0/
H A DIGnssBatching.hal92 * @return batchSize number of location objects supported per batch
94 getBatchSize() generates (uint16_t batchSize);

123