Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DLSTM.cpp383 bool timeMajor, bool forwardSequence) { in LSTMEvalFloat32() argument
431 inputData + (forwardSequence ? 0 : batchInputSize * (maxTime - 1)); in LSTMEvalFloat32()
433 hasAuxInput ? (auxInputData + (forwardSequence ? 0 : batchInputSize * (maxTime - 1))) in LSTMEvalFloat32()
436 outputData + (forwardSequence ? 0 : batchOutputSize * (maxTime - 1)); in LSTMEvalFloat32()
437 const int batchInputDelta = forwardSequence ? batchInputSize : -batchInputSize; in LSTMEvalFloat32()
438 const int batchOutputDelta = forwardSequence ? batchOutputSize : -batchOutputSize; in LSTMEvalFloat32()
503 bool timeMajor, bool forwardSequence) { in LSTMEvalFloat16()
669 inputData + (forwardSequence ? 0 : batchInputSize * (maxTime - 1)); in LSTMEvalFloat16()
671 hasAuxInput ? (auxInputData + (forwardSequence ? 0 : batchInputSize * (maxTime - 1))) in LSTMEvalFloat16()
674 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);