Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DUnidirectionalSequenceRNN.cpp91 const uint32_t numUnits = getSizeOfDimension(weightsShape, 0); in executeTyped() local
144 const uint32_t numUnits = getSizeOfDimension(weights, 0); in prepare() local
DRNN.cpp127 uint32_t numUnits = weightsShape.dimensions[0]; in RNNStep() local
/frameworks/ml/nn/runtime/test/
DTestValidateOperations.cpp2930 const uint32_t numUnits = 5; in bidirectionlSequenceRNNTest() local
3006 const uint32_t numUnits = 5; in unidirectionlSequenceRNNTest() local
3070 const uint32_t numUnits = 4; in unidirectionalSequenceLSTMTest() local