Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DUnidirectionalSequenceRNN.cpp99 const uint32_t numUnits = getSizeOfDimension(weightsShape, 0); in executeTyped() local
166 const uint32_t numUnits = getSizeOfDimension(weights, 0); in prepare() local
H A DRNN.cpp123 uint32_t numUnits = weightsShape.dimensions[0]; in RNNStep() local
H A DQLSTM.cpp189 const uint32_t numUnits = getSizeOfDimension(inputToOutputShape, 0); in prepare() local
394 const uint32_t numUnits = inputToOutputWeightsShape.dimensions[0]; in execute() local
/aosp12/packages/modules/NeuralNetworks/runtime/test/
H A DTestValidateOperations.cpp3928 const uint32_t numUnits = 5; in bidirectionlSequenceRNNTest() local
4004 const uint32_t numUnits = 5; in unidirectionlSequenceRNNTest() local
4068 const uint32_t numUnits = 4; in unidirectionalSequenceLSTMTest() local