Searched refs:batchOutputSize (Results 1 – 1 of 1) sorted by relevance
443 const uint32_t batchOutputSize = batchSize * outputSize; in LSTMEvalFloat32() local460 transposedOutput.resize(maxTime * batchOutputSize); in LSTMEvalFloat32()479 outputData + (forwardSequence ? 0 : batchOutputSize * (maxTime - 1)); in LSTMEvalFloat32()481 const int batchOutputDelta = (forwardSequence ? 1 : -1) * static_cast<int>(batchOutputSize); in LSTMEvalFloat32()563 const uint32_t batchOutputSize = batchSize * outputSize; in LSTMEvalFloat16() local669 std::vector<float> output_state_out_float32(batchOutputSize); in LSTMEvalFloat16()674 std::vector<float> output_float32(maxTime * batchOutputSize); in LSTMEvalFloat16()696 transposedOutput.resize(maxTime * batchOutputSize); in LSTMEvalFloat16()717 outputData + (forwardSequence ? 0 : batchOutputSize * (maxTime - 1)); in LSTMEvalFloat16()719 const int batchOutputDelta = (forwardSequence ? 1 : -1) * static_cast<int>(batchOutputSize); in LSTMEvalFloat16()