Searched refs:fwHiddenState (Results 1 – 3 of 3) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | BidirectionalSequenceRNN.cpp | 132 const T* fwHiddenState = context->getInputBuffer<T>(kFwHiddenStateTensor); in executeTyped() local
268 fixedTimeAuxInputShape, fwHiddenState, fwBias, fwWeights, fwWeightsShape, in executeTyped()
273 fwHiddenState = fwOutputHiddenState; in executeTyped()
372 Shape fwHiddenState = context->getInputShape(kFwHiddenStateTensor); in prepare() local
398 NN_RET_CHECK_EQ(getNumberOfDimensions(fwHiddenState), 2); in prepare()
408 NN_RET_CHECK_EQ(batchSize, getSizeOfDimension(fwHiddenState, 0)); in prepare()
409 NN_RET_CHECK_EQ(fwNumUnits, getSizeOfDimension(fwHiddenState, 1)); in prepare()
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 3961 ANeuralNetworksOperandType fwHiddenState = {.type = inputOperandCode, in bidirectionlSequenceRNNTest() local
3966 ANeuralNetworksOperandType bwHiddenState = fwHiddenState; in bidirectionlSequenceRNNTest()
3986 {input, fwWeights, fwRecurrentWeights, fwBias, fwHiddenState, in bidirectionlSequenceRNNTest()
|
| /frameworks/ml/nn/tools/api/ |
| D | types.spec | 3502 * * 4: fwHiddenState.
|