Searched refs:fwWeights (Results 1 – 3 of 3) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | BidirectionalSequenceRNN.cpp | 96 const T* fwWeights = context->getInputBuffer<T>(kFwWeightsTensor); in executeTyped() local
237 fixedTimeAuxInputShape, fwHiddenState, fwBias, fwWeights, fwWeightsShape, in executeTyped()
309 Shape fwWeights = context->getInputShape(kFwWeightsTensor); in prepare() local
330 const uint32_t fwNumUnits = getSizeOfDimension(fwWeights, 0); in prepare()
335 NN_RET_CHECK_EQ(getNumberOfDimensions(fwWeights), 2u); in prepare()
344 NN_RET_CHECK_EQ(inputSize, getSizeOfDimension(fwWeights, 1)); in prepare()
|
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | TestValidateOperations.cpp | 3981 ANeuralNetworksOperandType fwWeights = {.type = inputOperandCode, in bidirectionlSequenceRNNTest() local
3986 ANeuralNetworksOperandType bwWeights = fwWeights; in bidirectionlSequenceRNNTest()
4024 {input, fwWeights, fwRecurrentWeights, fwBias, fwHiddenState, in bidirectionlSequenceRNNTest()
4026 fwWeights, bwWeights, activation, timeMajor, mergeOutputs}, in bidirectionlSequenceRNNTest()
|
| /packages/modules/NeuralNetworks/tools/api/ |
| D | types.spec | 3659 * * 1: fwWeights.
|