Searched refs:fwNumUnits (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | BidirectionalSequenceRNN.cpp | 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()
[all …]
|
| /packages/modules/NeuralNetworks/tools/api/ |
| D | types.spec | 3660 * A 2-D tensor of shape [fwNumUnits, inputSize].
3662 * A 2-D tensor of shape [fwNumUnits, fwNumUnits].
3664 * A 1-D tensor of shape [fwNumUnits].
3666 * A 2-D tensor of shape [batchSize, fwNumUnits]. Specifies a hidden
3685 * A 2-D tensor of shape [fwNumUnits, auxInputSize]. Can be omitted.
3694 * A 2-D tensor of shape [fwNumUnits, inputSize].
3715 * dimension is equal to (fwNumUnits + bwNumUnits), otherwise it is set
3716 * to fwNumUnits.
3725 * A 2-D tensor of shape [batchSize, fwNumUnits] containing a hidden
|