Lines Matching refs:timeMajor
383 bool timeMajor, bool forwardSequence) { in LSTMEvalFloat32() argument
390 (inputRank == 3) ? getSizeOfDimension(input_shape, timeMajor ? 0 : 1) : 1; in LSTMEvalFloat32()
391 const uint32_t batchSize = (inputRank == 3) ? getSizeOfDimension(input_shape, timeMajor ? 1 : 0) in LSTMEvalFloat32()
408 if (!timeMajor) { in LSTMEvalFloat32()
421 const float* inputData = timeMajor ? input_buffer : transposedInput.data(); in LSTMEvalFloat32()
423 hasAuxInput ? (timeMajor ? aux_input_buffer : transposedAuxInput.data()) : nullptr; in LSTMEvalFloat32()
424 float* outputData = timeMajor ? output_buffer : transposedOutput.data(); in LSTMEvalFloat32()
469 if (!timeMajor) { in LSTMEvalFloat32()
503 bool timeMajor, bool forwardSequence) { in LSTMEvalFloat16()
510 (inputRank == 3) ? getSizeOfDimension(input_shape, timeMajor ? 0 : 1) : 1; in LSTMEvalFloat16()
511 const uint32_t batchSize = (inputRank == 3) ? getSizeOfDimension(input_shape, timeMajor ? 1 : 0) in LSTMEvalFloat16()
643 if (!timeMajor) { in LSTMEvalFloat16()
657 const float* inputData = timeMajor ? input_float32.data() : transposedInput.data(); in LSTMEvalFloat16()
659 hasAuxInput ? (timeMajor ? aux_input_float32.data() : transposedAuxInput.data()) in LSTMEvalFloat16()
661 float* outputData = timeMajor ? output_float32.data() : transposedOutput.data(); in LSTMEvalFloat16()
709 if (!timeMajor) { in LSTMEvalFloat16()