Searched refs:batchOutputSize (Results 1 – 1 of 1) sorted by relevance
448 const uint32_t batchOutputSize = batchSize * outputSize; in LSTMEvalFloat32() local465 transposedOutput.resize(maxTime * batchOutputSize); in LSTMEvalFloat32()484 outputData + (forwardSequence ? 0 : batchOutputSize * (maxTime - 1)); in LSTMEvalFloat32()486 const int batchOutputDelta = (forwardSequence ? 1 : -1) * static_cast<int>(batchOutputSize); in LSTMEvalFloat32()568 const uint32_t batchOutputSize = batchSize * outputSize; in LSTMEvalFloat16() local674 std::vector<float> output_state_out_float32(batchOutputSize); in LSTMEvalFloat16()679 std::vector<float> output_float32(maxTime * batchOutputSize); in LSTMEvalFloat16()701 transposedOutput.resize(maxTime * batchOutputSize); in LSTMEvalFloat16()722 outputData + (forwardSequence ? 0 : batchOutputSize * (maxTime - 1)); in LSTMEvalFloat16()724 const int batchOutputDelta = (forwardSequence ? 1 : -1) * static_cast<int>(batchOutputSize); in LSTMEvalFloat16()