Searched refs:transposedOutput (Results 1 – 1 of 1) sorted by relevance
448 std::vector<float> transposedOutput; in LSTMEvalFloat32() local460 transposedOutput.resize(maxTime * batchOutputSize); in LSTMEvalFloat32()467 float* outputData = timeMajor ? output_buffer : transposedOutput.data(); in LSTMEvalFloat32()513 transposeFirstTwoDimensions<float>(transposedOutput.data(), transposedOutputShape, in LSTMEvalFloat32()683 std::vector<float> transposedOutput; in LSTMEvalFloat16() local696 transposedOutput.resize(maxTime * batchOutputSize); in LSTMEvalFloat16()704 float* outputData = timeMajor ? output_float32.data() : transposedOutput.data(); in LSTMEvalFloat16()753 transposeFirstTwoDimensions<float>(transposedOutput.data(), transposedOutputShape, in LSTMEvalFloat16()