Lines Matching refs:timeMajor
426 bool timeMajor, bool forwardSequence) { in LSTMEvalFloat32() argument
433 (inputRank == 3) ? getSizeOfDimension(input_shape, timeMajor ? 0 : 1) : 1; in LSTMEvalFloat32()
434 const uint32_t batchSize = (inputRank == 3) ? getSizeOfDimension(input_shape, timeMajor ? 1 : 0) in LSTMEvalFloat32()
451 if (!timeMajor) { in LSTMEvalFloat32()
464 const float* inputData = timeMajor ? input_buffer : transposedInput.data(); in LSTMEvalFloat32()
466 hasAuxInput ? (timeMajor ? aux_input_buffer : transposedAuxInput.data()) : nullptr; in LSTMEvalFloat32()
467 float* outputData = timeMajor ? output_buffer : transposedOutput.data(); in LSTMEvalFloat32()
512 if (!timeMajor) { in LSTMEvalFloat32()
546 bool timeMajor, bool forwardSequence) { in LSTMEvalFloat16()
553 (inputRank == 3) ? getSizeOfDimension(input_shape, timeMajor ? 0 : 1) : 1; in LSTMEvalFloat16()
554 const uint32_t batchSize = (inputRank == 3) ? getSizeOfDimension(input_shape, timeMajor ? 1 : 0) in LSTMEvalFloat16()
686 if (!timeMajor) { in LSTMEvalFloat16()
700 const float* inputData = timeMajor ? input_float32.data() : transposedInput.data(); in LSTMEvalFloat16()
702 hasAuxInput ? (timeMajor ? aux_input_float32.data() : transposedAuxInput.data()) in LSTMEvalFloat16()
704 float* outputData = timeMajor ? output_float32.data() : transposedOutput.data(); in LSTMEvalFloat16()
752 if (!timeMajor) { in LSTMEvalFloat16()