Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DQLSTM.cpp188 const uint32_t numUnits = getSizeOfDimension(inputToOutputShape, 0); in prepare() local
192 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToOutputShape, 0), numUnits); in prepare()
198 NN_RET_CHECK_EQ(getSizeOfDimension(inputToInputShape, 0), numUnits); in prepare()
204 NN_RET_CHECK_EQ(getSizeOfDimension(inputToForgetShape, 0), numUnits); in prepare()
208 NN_RET_CHECK_EQ(getSizeOfDimension(inputToCellShape, 0), numUnits); in prepare()
214 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToInputShape, 0), numUnits); in prepare()
220 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToForgetShape, 0), numUnits); in prepare()
224 NN_RET_CHECK_EQ(getSizeOfDimension(recurrentToCellShape, 0), numUnits); in prepare()
238 NN_RET_CHECK_EQ(getSizeOfDimension(cellToInputShape, 0), numUnits); in prepare()
244 NN_RET_CHECK_EQ(getSizeOfDimension(cellToForgetShape, 0), numUnits); in prepare()
[all …]
DUnidirectionalSequenceRNN.cpp100 const uint32_t numUnits = getSizeOfDimension(weightsShape, 0); in executeTyped() local
113 output += batchSize * numUnits; in executeTyped()
124 std::copy(hiddenState, hiddenState + batchSize * numUnits, stateOutput); in executeTyped()
166 const uint32_t numUnits = getSizeOfDimension(weights, 0); in prepare() local
176 NN_RET_CHECK_EQ(numUnits, getSizeOfDimension(bias, 0)); in prepare()
177 NN_RET_CHECK_EQ(numUnits, getSizeOfDimension(recurrentWeights, 0)); in prepare()
178 NN_RET_CHECK_EQ(numUnits, getSizeOfDimension(recurrentWeights, 1)); in prepare()
180 NN_RET_CHECK_EQ(numUnits, getSizeOfDimension(hiddenState, 1)); in prepare()
186 output.dimensions[2] = numUnits; in prepare()
193 outputStateShape.dimensions[1] = numUnits; in prepare()
DRNN.cpp127 uint32_t numUnits = weightsShape.dimensions[0]; in RNNStep() local
132 /*outputBatchStride=*/numUnits, /*outputBatchOffset=*/0, outputData); in RNNStep()
/frameworks/ml/nn/runtime/test/
DTestValidateOperations.cpp3928 const uint32_t numUnits = 5; in bidirectionlSequenceRNNTest() local
3931 uint32_t weightsDims[2] = {inputSize, numUnits}; in bidirectionlSequenceRNNTest()
3932 uint32_t recurrentWeightsDims[2] = {numUnits, numUnits}; in bidirectionlSequenceRNNTest()
3933 uint32_t biasDims[1] = {numUnits}; in bidirectionlSequenceRNNTest()
3934 uint32_t hiddenStateDims[2] = {batchSize, numUnits}; in bidirectionlSequenceRNNTest()
3935 uint32_t outputDims[2] = {batchSize, numUnits}; in bidirectionlSequenceRNNTest()
4004 const uint32_t numUnits = 5; in unidirectionlSequenceRNNTest() local
4007 uint32_t weightsDims[2] = {inputSize, numUnits}; in unidirectionlSequenceRNNTest()
4008 uint32_t recurrentWeightsDims[2] = {numUnits, numUnits}; in unidirectionlSequenceRNNTest()
4009 uint32_t biasDims[1] = {numUnits}; in unidirectionlSequenceRNNTest()
[all …]