Searched refs:fwHiddenState (Results 1 – 3 of 3) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | BidirectionalSequenceRNN.cpp | 101 const T* fwHiddenState = context->getInputBuffer<T>(kFwHiddenStateTensor); in executeTyped() local
237 fixedTimeAuxInputShape, fwHiddenState, fwBias, fwWeights, fwWeightsShape, in executeTyped()
242 fwHiddenState = fwOutputHiddenState; in executeTyped()
312 Shape fwHiddenState = context->getInputShape(kFwHiddenStateTensor); in prepare() local
338 NN_RET_CHECK_EQ(getNumberOfDimensions(fwHiddenState), 2u); in prepare()
348 NN_RET_CHECK_EQ(batchSize, getSizeOfDimension(fwHiddenState, 0)); in prepare()
349 NN_RET_CHECK_EQ(fwNumUnits, getSizeOfDimension(fwHiddenState, 1)); in prepare()
|
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | TestValidateOperations.cpp | 3999 ANeuralNetworksOperandType fwHiddenState = {.type = inputOperandCode, in bidirectionlSequenceRNNTest() local
4004 ANeuralNetworksOperandType bwHiddenState = fwHiddenState; in bidirectionlSequenceRNNTest()
4024 {input, fwWeights, fwRecurrentWeights, fwBias, fwHiddenState, in bidirectionlSequenceRNNTest()
|
| /packages/modules/NeuralNetworks/tools/api/ |
| D | types.spec | 3665 * * 4: fwHiddenState.
|