Searched refs:transposeFirstTwoDims (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | UnidirectionalSequenceRNN.cpp | 37 void transposeFirstTwoDims(const T* input, const Shape& inputShape, T* output) { in transposeFirstTwoDims() function
77 transposeFirstTwoDims(input, inputShape, inputTransposed.data()); in executeTyped()
104 transposeFirstTwoDims(outputTransposed.data(), outputShape, in executeTyped()
|
| D | BidirectionalSequenceRNN.cpp | 36 void transposeFirstTwoDims(const T* input, const Shape& inputShape, T* output) { in transposeFirstTwoDims() function
161 transposeFirstTwoDims(input, inputShape, inputTransposed.data()); in executeTyped()
163 transposeFirstTwoDims(auxInput, auxInputShape, auxInputTransposed.data()); in executeTyped()
276 transposeFirstTwoDims(fwOutputTransposed.data(), fwOutputShape, in executeTyped()
279 transposeFirstTwoDims(bwOutputTransposed.data(), bwOutputShape, in executeTyped()
|