Home
last modified time | relevance | path

Searched refs:fwOutputBatchStride (Results 1 – 1 of 1) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DBidirectionalSequenceRNN.cpp262 const uint32_t fwOutputBatchStride = mergeOutputs ? (fwNumUnits + bwNumUnits) : fwNumUnits; in executeTyped() local
263 T* fwOutputBatchPtr = fwOutput + i * batchSize * fwOutputBatchStride; in executeTyped()
268 fwRecurrentWeightsShape, activation, fwOutputBatchStride, in executeTyped()