Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DUnidirectionalSequenceRNN.cpp39 const uint32_t secondDimSize = getSizeOfDimension(inputShape, 1); in transposeFirstTwoDims() local
42 for (uint32_t s = 0; s < secondDimSize; ++s) { in transposeFirstTwoDims()
44 const uint32_t inputIndex = f * secondDimSize * inputSize + s * inputSize + i; in transposeFirstTwoDims()
DBidirectionalSequenceRNN.cpp38 const uint32_t secondDimSize = getSizeOfDimension(inputShape, 1); in transposeFirstTwoDims() local
41 for (uint32_t s = 0; s < secondDimSize; ++s) { in transposeFirstTwoDims()
43 const uint32_t inputIndex = f * secondDimSize * inputSize + s * inputSize + i; in transposeFirstTwoDims()