Searched refs:transposedOutput (Results 1 – 1 of 1) sorted by relevance
405 std::vector<float> transposedOutput; in LSTMEvalFloat32() local417 transposedOutput.resize(maxTime * batchOutputSize); in LSTMEvalFloat32()424 float* outputData = timeMajor ? output_buffer : transposedOutput.data(); in LSTMEvalFloat32()470 transposeFirstTwoDimensions<float>(transposedOutput.data(), transposedOutputShape, in LSTMEvalFloat32()640 std::vector<float> transposedOutput; in LSTMEvalFloat16() local653 transposedOutput.resize(maxTime * batchOutputSize); in LSTMEvalFloat16()661 float* outputData = timeMajor ? output_float32.data() : transposedOutput.data(); in LSTMEvalFloat16()710 transposeFirstTwoDimensions<float>(transposedOutput.data(), transposedOutputShape, in LSTMEvalFloat16()