Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DBidirectionalSequenceRNN.cpp191 uint32_t auxInputSize = 0; in executeTyped() local
193 auxInputSize = getSizeOfDimension(auxInputShape, 2); in executeTyped()
231 auxInputBatchPtr = auxInput + i * batchSize * auxInputSize; in executeTyped()
250 auxInputBatchPtr = auxInput + i * batchSize * auxInputSize; in executeTyped()
/packages/modules/NeuralNetworks/tools/api/
Dtypes.spec3681 * auxInputSize], otherwise the input has a shape [batchSize, maxTime,
3682 * auxInputSize]. Can be omitted. See the docs above for the usage
3685 * A 2-D tensor of shape [fwNumUnits, auxInputSize]. Can be omitted.
3688 * A 2-D tensor of shape [bwNumUnits, auxInputSize]. Can be omitted.