Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DLSTM.cpp480 const int batchInputDelta = (forwardSequence ? 1 : -1) * static_cast<int>(batchInputSize); in LSTMEvalFloat32() local
501 inputCurrentTimeStep += batchInputDelta; in LSTMEvalFloat32()
503 auxInputCurrentTimeStep += batchInputDelta; in LSTMEvalFloat32()
718 const int batchInputDelta = (forwardSequence ? 1 : -1) * static_cast<int>(batchInputSize); in LSTMEvalFloat16() local
743 inputCurrentTimeStep += batchInputDelta; in LSTMEvalFloat16()
745 auxInputCurrentTimeStep += batchInputDelta; in LSTMEvalFloat16()