Searched refs:transposedInput (Results 1 – 1 of 1) sorted by relevance
402 std::vector<float> transposedInput; in LSTMEvalFloat32() local409 transposedInput.resize(maxTime * batchInputSize); in LSTMEvalFloat32()410 transposeFirstTwoDimensions<float>(input_buffer, input_shape, transposedInput.data()); in LSTMEvalFloat32()421 const float* inputData = timeMajor ? input_buffer : transposedInput.data(); in LSTMEvalFloat32()637 std::vector<float> transposedInput; in LSTMEvalFloat16() local644 transposedInput.resize(maxTime * batchInputSize); in LSTMEvalFloat16()646 transposedInput.data()); in LSTMEvalFloat16()657 const float* inputData = timeMajor ? input_float32.data() : transposedInput.data(); in LSTMEvalFloat16()