Searched refs:transposeFirstTwoDims (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | UnidirectionalSequenceRNN.cpp | 41 void transposeFirstTwoDims(const T* input, const Shape& inputShape, T* output) { in transposeFirstTwoDims() function
81 transposeFirstTwoDims(input, inputShape, inputTransposed.data()); in executeTyped()
108 transposeFirstTwoDims(outputTransposed.data(), outputShape, in executeTyped()
|
| D | BidirectionalSequenceRNN.cpp | 53 void transposeFirstTwoDims(const T* input, const Shape& inputShape, T* output) { in transposeFirstTwoDims() function
141 transposeFirstTwoDims(input, inputShape, inputTransposed.data()); in executeTyped()
143 transposeFirstTwoDims(auxInput, auxInputShape, auxInputTransposed.data()); in executeTyped()
237 transposeFirstTwoDims(fwOutputTransposed.data(), fwOutputShape, in executeTyped()
240 transposeFirstTwoDims(bwOutputTransposed.data(), bwOutputShape, in executeTyped()
|