Searched refs:bwHiddenState (Results 1 – 3 of 3) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | BidirectionalSequenceRNN.cpp | 139 const T* bwHiddenState = context->getInputBuffer<T>(kBwHiddenStateTensor); in executeTyped() local
296 fixedTimeAuxInputShape, bwHiddenState, bwBias, bwWeights, bwWeightsShape, in executeTyped()
301 bwHiddenState = bwOutputHiddenState; in executeTyped()
376 Shape bwHiddenState = context->getInputShape(kBwHiddenStateTensor); in prepare() local
402 NN_RET_CHECK_EQ(getNumberOfDimensions(bwHiddenState), 2); in prepare()
417 NN_RET_CHECK_EQ(batchSize, getSizeOfDimension(bwHiddenState, 0)); in prepare()
418 NN_RET_CHECK_EQ(bwNumUnits, getSizeOfDimension(bwHiddenState, 1)); in prepare()
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 3966 ANeuralNetworksOperandType bwHiddenState = fwHiddenState; in bidirectionlSequenceRNNTest() local
3987 bwWeights, bwRecurrentWeights, bwBias, bwHiddenState, input, in bidirectionlSequenceRNNTest()
|
| /frameworks/ml/nn/tools/api/ |
| D | types.spec | 3511 * * 8: bwHiddenState
|