Searched refs:bwWeights (Results 1 – 3 of 3) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | BidirectionalSequenceRNN.cpp | 103 const T* bwWeights = context->getInputBuffer<T>(kBwWeightsTensor); in executeTyped() local
265 fixedTimeAuxInputShape, bwHiddenState, bwBias, bwWeights, bwWeightsShape, in executeTyped()
313 Shape bwWeights = context->getInputShape(kBwWeightsTensor); in prepare() local
331 const uint32_t bwNumUnits = getSizeOfDimension(bwWeights, 0); in prepare()
339 NN_RET_CHECK_EQ(getNumberOfDimensions(bwWeights), 2u); in prepare()
352 NN_RET_CHECK_EQ(inputSize, getSizeOfDimension(bwWeights, 1)); in prepare()
376 NN_RET_CHECK_EQ(getSizeOfDimension(auxInput, 2), getSizeOfDimension(bwWeights, 1)); in prepare()
|
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | TestValidateOperations.cpp | 3986 ANeuralNetworksOperandType bwWeights = fwWeights; in bidirectionlSequenceRNNTest() local
4025 bwWeights, bwRecurrentWeights, bwBias, bwHiddenState, input, in bidirectionlSequenceRNNTest()
4026 fwWeights, bwWeights, activation, timeMajor, mergeOutputs}, in bidirectionlSequenceRNNTest()
|
| /packages/modules/NeuralNetworks/tools/api/ |
| D | types.spec | 3668 * * 5: bwWeights.
|