Home
last modified time | relevance | path

Searched refs:inputTransposed (Results 1 – 2 of 2) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DUnidirectionalSequenceRNN.cpp83 std::vector<T> inputTransposed; in executeTyped() local
87 inputTransposed.resize(getNumberOfElements(inputShape)); in executeTyped()
89 transposeFirstTwoDims(input, inputShape, inputTransposed.data()); in executeTyped()
90 input = inputTransposed.data(); in executeTyped()
H A DBidirectionalSequenceRNN.cpp174 std::vector<T> inputTransposed; in executeTyped() local
180 inputTransposed.resize(getNumberOfElements(inputShape)); in executeTyped()
190 transposeFirstTwoDims(input, inputShape, inputTransposed.data()); in executeTyped()
196 input = inputTransposed.data(); in executeTyped()