Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DLSTM.cpp426 bool timeMajor, bool forwardSequence) { in LSTMEvalFloat32() argument
474 inputData + (forwardSequence ? 0 : batchInputSize * (maxTime - 1)); in LSTMEvalFloat32()
476 hasAuxInput ? (auxInputData + (forwardSequence ? 0 : batchInputSize * (maxTime - 1))) in LSTMEvalFloat32()
479 outputData + (forwardSequence ? 0 : batchOutputSize * (maxTime - 1)); in LSTMEvalFloat32()
480 const int batchInputDelta = forwardSequence ? batchInputSize : -batchInputSize; in LSTMEvalFloat32()
481 const int batchOutputDelta = forwardSequence ? batchOutputSize : -batchOutputSize; in LSTMEvalFloat32()
546 bool timeMajor, bool forwardSequence) { in LSTMEvalFloat16()
712 inputData + (forwardSequence ? 0 : batchInputSize * (maxTime - 1)); in LSTMEvalFloat16()
714 hasAuxInput ? (auxInputData + (forwardSequence ? 0 : batchInputSize * (maxTime - 1))) in LSTMEvalFloat16()
717 outputData + (forwardSequence ? 0 : batchOutputSize * (maxTime - 1)); in LSTMEvalFloat16()
[all …]
DLSTM.h130 bool timeMajor = true, bool forwardSequence = true);
158 _Float16* scratch_buffer_buffer, bool timeMajor = true, bool forwardSequence = true);