Searched refs:transposeFirstTwoDims (Results 1 – 2 of 2) sorted by relevance
| /aosp12/packages/modules/NeuralNetworks/common/operations/ |
| H A D | UnidirectionalSequenceRNN.cpp | 49 void transposeFirstTwoDims(const T* input, const Shape& inputShape, T* output) { in transposeFirstTwoDims() function
89 transposeFirstTwoDims(input, inputShape, inputTransposed.data()); in executeTyped()
116 transposeFirstTwoDims(outputTransposed.data(), outputShape, in executeTyped()
|
| H A D | BidirectionalSequenceRNN.cpp | 65 void transposeFirstTwoDims(const T* input, const Shape& inputShape, T* output) { in transposeFirstTwoDims() function
190 transposeFirstTwoDims(input, inputShape, inputTransposed.data()); in executeTyped()
192 transposeFirstTwoDims(auxInput, auxInputShape, auxInputTransposed.data()); in executeTyped()
305 transposeFirstTwoDims(fwOutputTransposed.data(), fwOutputShape, in executeTyped()
308 transposeFirstTwoDims(bwOutputTransposed.data(), bwOutputShape, in executeTyped()
|