Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DUnidirectionalSequenceRNN.cpp38 const uint32_t firstDimSize = getSizeOfDimension(inputShape, 0); in transposeFirstTwoDims() local
41 for (uint32_t f = 0; f < firstDimSize; ++f) { in transposeFirstTwoDims()
45 const uint32_t outputIndex = s * firstDimSize * inputSize + f * inputSize + i; in transposeFirstTwoDims()
DBidirectionalSequenceRNN.cpp37 const uint32_t firstDimSize = getSizeOfDimension(inputShape, 0); in transposeFirstTwoDims() local
40 for (uint32_t f = 0; f < firstDimSize; ++f) { in transposeFirstTwoDims()
44 const uint32_t outputIndex = s * firstDimSize * inputSize + f * inputSize + i; in transposeFirstTwoDims()