Searched refs:bwNumUnits (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | BidirectionalSequenceRNN.cpp | 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()
[all …]
|
| /packages/modules/NeuralNetworks/tools/api/ |
| D | types.spec | 3669 * A 2-D tensor of shape [bwNumUnits, inputSize].
3671 * A 2-D tensor of shape [bwNumUnits, bwNumUnits].
3673 * A 1-D tensor of shape [bwNumUnits].
3675 * A 2-D tensor of shape [batchSize, bwNumUnits]. Specifies a hidden
3688 * A 2-D tensor of shape [bwNumUnits, auxInputSize]. Can be omitted.
3696 * A 2-D tensor of shape [bwNumUnits, inputSize].
3715 * dimension is equal to (fwNumUnits + bwNumUnits), otherwise it is set
3721 * [maxTime, batchSize, bwNumUnits], otherwise the shape is set to
3722 * [batchSize, maxTime, bwNumUnits].
3731 * A 2-D tensor of shape [batchSize, bwNumUnits] containing a hidden
|