Searched refs:transposedInput (Results 1 – 1 of 1) sorted by relevance
445 std::vector<float> transposedInput; in LSTMEvalFloat32() local452 transposedInput.resize(maxTime * batchInputSize); in LSTMEvalFloat32()453 transposeFirstTwoDimensions<float>(input_buffer, input_shape, transposedInput.data()); in LSTMEvalFloat32()464 const float* inputData = timeMajor ? input_buffer : transposedInput.data(); in LSTMEvalFloat32()680 std::vector<float> transposedInput; in LSTMEvalFloat16() local687 transposedInput.resize(maxTime * batchInputSize); in LSTMEvalFloat16()689 transposedInput.data()); in LSTMEvalFloat16()700 const float* inputData = timeMajor ? input_float32.data() : transposedInput.data(); in LSTMEvalFloat16()