Lines Matching refs:bwNumUnits
196 const uint32_t bwNumUnits = getSizeOfDimension(bwWeightsShape, 0); in executeTyped() local
221 tempHiddenState.resize(std::max(batchSize * fwNumUnits, batchSize * bwNumUnits)); in executeTyped()
233 const uint32_t fwOutputBatchStride = mergeOutputs ? (fwNumUnits + bwNumUnits) : fwNumUnits; in executeTyped()
256 bwOutputBatchStride = fwNumUnits + bwNumUnits; in executeTyped()
260 bwOutputBatchStride = bwNumUnits; in executeTyped()
331 const uint32_t bwNumUnits = getSizeOfDimension(bwWeights, 0); in prepare() local
354 NN_RET_CHECK_EQ(bwNumUnits, getSizeOfDimension(bwBias, 0)); in prepare()
355 NN_RET_CHECK_EQ(bwNumUnits, getSizeOfDimension(bwRecurrentWeights, 0)); in prepare()
356 NN_RET_CHECK_EQ(bwNumUnits, getSizeOfDimension(bwRecurrentWeights, 1)); in prepare()
358 NN_RET_CHECK_EQ(bwNumUnits, getSizeOfDimension(bwHiddenState, 1)); in prepare()
369 NN_RET_CHECK_EQ(getSizeOfDimension(bwAuxWeights, 0), bwNumUnits); in prepare()
383 fwOutput.dimensions[2] = mergeOutputs ? fwNumUnits + bwNumUnits : fwNumUnits; in prepare()
390 bwOutput.dimensions[2] = bwNumUnits; in prepare()