Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DLSTM.cpp480 const int batchInputDelta = forwardSequence ? batchInputSize : -batchInputSize; in LSTMEvalFloat32() local
501 inputCurrentTimeStep += batchInputDelta; in LSTMEvalFloat32()
503 auxInputCurrentTimeStep += batchInputDelta; in LSTMEvalFloat32()
718 const int batchInputDelta = forwardSequence ? batchInputSize : -batchInputSize; in LSTMEvalFloat16() local
743 inputCurrentTimeStep += batchInputDelta; in LSTMEvalFloat16()
745 auxInputCurrentTimeStep += batchInputDelta; in LSTMEvalFloat16()