Searched refs:bwNumUnits (Results 1 – 1 of 1) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | BidirectionalSequenceRNN.cpp | 176 const uint32_t bwNumUnits = getSizeOfDimension(bwWeightsShape, 0); in executeTyped() local
193 const uint32_t fwOutputBatchStride = mergeOutputs ? (fwNumUnits + bwNumUnits) : fwNumUnits; in executeTyped()
205 tempHiddenState.resize(batchSize * bwNumUnits); in executeTyped()
217 bwOutputBatchStride = fwNumUnits + bwNumUnits; in executeTyped()
221 bwOutputBatchStride = bwNumUnits; in executeTyped()
320 const uint32_t bwNumUnits = getSizeOfDimension(bwWeights, 0); in prepare() local
341 NN_RET_CHECK_EQ(bwNumUnits, getSizeOfDimension(bwBias, 0)); in prepare()
342 NN_RET_CHECK_EQ(bwNumUnits, getSizeOfDimension(bwRecurrentWeights, 0)); in prepare()
343 NN_RET_CHECK_EQ(bwNumUnits, getSizeOfDimension(bwRecurrentWeights, 1)); in prepare()
345 NN_RET_CHECK_EQ(bwNumUnits, getSizeOfDimension(bwHiddenState, 1)); in prepare()
[all …]
|