Searched refs:bwHiddenState (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | BidirectionalSequenceRNN.cpp | 94 const T* bwHiddenState = context->getInputBuffer<T>(kBwHiddenStateTensor); in executeTyped() local
226 fixedTimeAuxInputShape, bwHiddenState, bwBias, bwWeights, bwWeightsShape, in executeTyped()
231 bwHiddenState = tempHiddenState.data(); in executeTyped()
299 Shape bwHiddenState = context->getInputShape(kBwHiddenStateTensor); in prepare() local
331 NN_RET_CHECK_EQ(getNumberOfDimensions(bwHiddenState), 2); in prepare()
344 NN_RET_CHECK_EQ(batchSize, getSizeOfDimension(bwHiddenState, 0)); in prepare()
345 NN_RET_CHECK_EQ(bwNumUnits, getSizeOfDimension(bwHiddenState, 1)); in prepare()
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 2967 ANeuralNetworksOperandType bwHiddenState = fwHiddenState; in bidirectionlSequenceRNNTest() local
2988 bwWeights, bwRecurrentWeights, bwBias, bwHiddenState, input, in bidirectionlSequenceRNNTest()
|