Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DLSTM.cpp437 const int batchInputDelta = forwardSequence ? batchInputSize : -batchInputSize; in LSTMEvalFloat32() local
458 inputCurrentTimeStep += batchInputDelta; in LSTMEvalFloat32()
460 auxInputCurrentTimeStep += batchInputDelta; in LSTMEvalFloat32()
675 const int batchInputDelta = forwardSequence ? batchInputSize : -batchInputSize; in LSTMEvalFloat16() local
700 inputCurrentTimeStep += batchInputDelta; in LSTMEvalFloat16()
702 auxInputCurrentTimeStep += batchInputDelta; in LSTMEvalFloat16()