Searched refs:bwHiddenState (Results 1 – 3 of 3) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | BidirectionalSequenceRNN.cpp | 108 const T* bwHiddenState = context->getInputBuffer<T>(kBwHiddenStateTensor); in executeTyped() local
265 fixedTimeAuxInputShape, bwHiddenState, bwBias, bwWeights, bwWeightsShape, in executeTyped()
270 bwHiddenState = bwOutputHiddenState; in executeTyped()
316 Shape bwHiddenState = context->getInputShape(kBwHiddenStateTensor); in prepare() local
342 NN_RET_CHECK_EQ(getNumberOfDimensions(bwHiddenState), 2u); in prepare()
357 NN_RET_CHECK_EQ(batchSize, getSizeOfDimension(bwHiddenState, 0)); in prepare()
358 NN_RET_CHECK_EQ(bwNumUnits, getSizeOfDimension(bwHiddenState, 1)); in prepare()
|
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | TestValidateOperations.cpp | 4004 ANeuralNetworksOperandType bwHiddenState = fwHiddenState; in bidirectionlSequenceRNNTest() local
4025 bwWeights, bwRecurrentWeights, bwBias, bwHiddenState, input, in bidirectionlSequenceRNNTest()
|
| /packages/modules/NeuralNetworks/tools/api/ |
| D | types.spec | 3674 * * 8: bwHiddenState
|