Lines Matching refs:fwNumUnits
195 const uint32_t fwNumUnits = getSizeOfDimension(fwWeightsShape, 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()
257 bwOutputBatchOffset = fwNumUnits; in executeTyped()
330 const uint32_t fwNumUnits = getSizeOfDimension(fwWeights, 0); in prepare() local
345 NN_RET_CHECK_EQ(fwNumUnits, getSizeOfDimension(fwBias, 0)); in prepare()
346 NN_RET_CHECK_EQ(fwNumUnits, getSizeOfDimension(fwRecurrentWeights, 0)); in prepare()
347 NN_RET_CHECK_EQ(fwNumUnits, getSizeOfDimension(fwRecurrentWeights, 1)); in prepare()
349 NN_RET_CHECK_EQ(fwNumUnits, getSizeOfDimension(fwHiddenState, 1)); in prepare()
367 NN_RET_CHECK_EQ(getSizeOfDimension(fwAuxWeights, 0), fwNumUnits); in prepare()
383 fwOutput.dimensions[2] = mergeOutputs ? fwNumUnits + bwNumUnits : fwNumUnits; in prepare()