Searched refs:transposeFirstTwoDimensions (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | LSTM.cpp | 453 transposeFirstTwoDimensions<float>(input_buffer, input_shape, transposedInput.data()); in LSTMEvalFloat32()
456 transposeFirstTwoDimensions<float>(aux_input_buffer, input_shape, in LSTMEvalFloat32()
459 transposeFirstTwoDimensions(input_shape, &transposedInputShape); in LSTMEvalFloat32()
513 transposeFirstTwoDimensions<float>(transposedOutput.data(), transposedOutputShape, in LSTMEvalFloat32()
688 transposeFirstTwoDimensions<float>(input_float32.data(), input_shape, in LSTMEvalFloat16()
692 transposeFirstTwoDimensions<float>(aux_input_float32.data(), input_shape, in LSTMEvalFloat16()
695 transposeFirstTwoDimensions(input_shape, &transposedInputShape); in LSTMEvalFloat16()
753 transposeFirstTwoDimensions<float>(transposedOutput.data(), transposedOutputShape, in LSTMEvalFloat16()
|
| /frameworks/ml/nn/common/include/ |
| D | OperationsUtils.h | 342 inline bool transposeFirstTwoDimensions(const T* buffer, const Shape& shape, T* transposedBuffer) { in transposeFirstTwoDimensions() function
363 inline bool transposeFirstTwoDimensions(const Shape& shape, Shape* transposedShape) { in transposeFirstTwoDimensions() function
|