Home
last modified time | relevance | path

Searched defs:numUnits (Results 1 – 4 of 4) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DUnidirectionalSequenceRNN.cpp87 const uint32_t numUnits = getSizeOfDimension(weightsShape, 0); in executeTyped() local
131 const uint32_t numUnits = getSizeOfDimension(weights, 0); in prepare() local
DQLSTM.cpp73 const uint32_t numUnits = getSizeOfDimension(inputToOutputShape, 0); in prepare() local
278 const uint32_t numUnits = inputToOutputWeightsShape.dimensions[0]; in execute() local
DRNN.cpp123 uint32_t numUnits = weightsShape.dimensions[0]; in RNNStep() local
/packages/modules/NeuralNetworks/runtime/test/
DTestValidateOperations.cpp3966 const uint32_t numUnits = 5; in bidirectionlSequenceRNNTest() local
4042 const uint32_t numUnits = 5; in unidirectionlSequenceRNNTest() local
4106 const uint32_t numUnits = 4; in unidirectionalSequenceLSTMTest() local