Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_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 ? 1 : -1) * static_cast<int>(batchInputSize); in LSTMEvalFloat32()
481 const int batchOutputDelta = (forwardSequence ? 1 : -1) * static_cast<int>(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 …]
/packages/modules/NeuralNetworks/common/types/operations/include/
DLSTM.h129 bool timeMajor = true, bool forwardSequence = true);
157 _Float16* scratch_buffer_buffer, bool timeMajor = true, bool forwardSequence = true);