Searched refs:transposedOutputShape (Results 1 – 1 of 1) sorted by relevance
450 Shape transposedOutputShape; in LSTMEvalFloat32() local461 transposedOutputShape = transposedInputShape; in LSTMEvalFloat32()462 transposedOutputShape.dimensions[2] = outputSize; in LSTMEvalFloat32()513 transposeFirstTwoDimensions<float>(transposedOutput.data(), transposedOutputShape, in LSTMEvalFloat32()685 Shape transposedOutputShape; in LSTMEvalFloat16() local697 transposedOutputShape = transposedInputShape; in LSTMEvalFloat16()698 transposedOutputShape.dimensions[2] = outputSize; in LSTMEvalFloat16()753 transposeFirstTwoDimensions<float>(transposedOutput.data(), transposedOutputShape, in LSTMEvalFloat16()