Searched refs:bwRecurrentWeights (Results 1 – 3 of 3) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | BidirectionalSequenceRNN.cpp | 105 const T* bwRecurrentWeights = context->getInputBuffer<T>(kBwRecurrentWeightsTensor); in executeTyped() local
266 bwAuxWeights, bwAuxWeightsShape, bwRecurrentWeights, in executeTyped()
314 Shape bwRecurrentWeights = context->getInputShape(kBwRecurrentWeightsTensor); in prepare() local
340 NN_RET_CHECK_EQ(getNumberOfDimensions(bwRecurrentWeights), 2u); in prepare()
355 NN_RET_CHECK_EQ(bwNumUnits, getSizeOfDimension(bwRecurrentWeights, 0)); in prepare()
356 NN_RET_CHECK_EQ(bwNumUnits, getSizeOfDimension(bwRecurrentWeights, 1)); in prepare()
|
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | TestValidateOperations.cpp | 3992 ANeuralNetworksOperandType bwRecurrentWeights = fwRecurrentWeights; in bidirectionlSequenceRNNTest() local
4025 bwWeights, bwRecurrentWeights, bwBias, bwHiddenState, input, in bidirectionlSequenceRNNTest()
|
| /packages/modules/NeuralNetworks/tools/api/ |
| D | types.spec | 3670 * * 6: bwRecurrentWeights.
|