Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DUnidirectionalSequenceRNN.cpp100 const uint32_t numUnits = getSizeOfDimension(weightsShape, 0); in executeTyped() local
166 const uint32_t numUnits = getSizeOfDimension(weights, 0); in prepare() local
DRNN.cpp127 uint32_t numUnits = weightsShape.dimensions[0]; in RNNStep() local
DQLSTM.cpp188 const uint32_t numUnits = getSizeOfDimension(inputToOutputShape, 0); in prepare() local
392 const uint32_t numUnits = inputToOutputWeightsShape.dimensions[0]; in execute() local
/frameworks/ml/nn/runtime/test/
DTestValidateOperations.cpp3929 const uint32_t numUnits = 5; in bidirectionlSequenceRNNTest() local
4005 const uint32_t numUnits = 5; in unidirectionlSequenceRNNTest() local
4069 const uint32_t numUnits = 4; in unidirectionalSequenceLSTMTest() local