Searched refs:transposeFirstTwoDimensions (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | LSTM.cpp | 410 transposeFirstTwoDimensions<float>(input_buffer, input_shape, transposedInput.data()); in LSTMEvalFloat32()
413 transposeFirstTwoDimensions<float>(aux_input_buffer, input_shape, in LSTMEvalFloat32()
416 transposeFirstTwoDimensions(input_shape, &transposedInputShape); in LSTMEvalFloat32()
470 transposeFirstTwoDimensions<float>(transposedOutput.data(), transposedOutputShape, in LSTMEvalFloat32()
645 transposeFirstTwoDimensions<float>(input_float32.data(), input_shape, in LSTMEvalFloat16()
649 transposeFirstTwoDimensions<float>(aux_input_float32.data(), input_shape, in LSTMEvalFloat16()
652 transposeFirstTwoDimensions(input_shape, &transposedInputShape); in LSTMEvalFloat16()
710 transposeFirstTwoDimensions<float>(transposedOutput.data(), transposedOutputShape, in LSTMEvalFloat16()
|
| /frameworks/ml/nn/common/include/ |
| D | OperationsUtils.h | 375 inline bool transposeFirstTwoDimensions(const T* buffer, const Shape& shape, T* transposedBuffer) { in transposeFirstTwoDimensions() function
396 inline bool transposeFirstTwoDimensions(const Shape& shape, Shape* transposedShape) { in transposeFirstTwoDimensions() function
|