Searched refs:hiddenState (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | UnidirectionalSequenceRNN.cpp | 74 const T* hiddenState = context->getInputBuffer<T>(kHiddenStateTensor); in executeTyped() local
109 RNN::RNNStep<T>(input, fixedTimeInputShape, hiddenState, bias, weights, weightsShape, in executeTyped()
112 hiddenState = output; in executeTyped()
124 std::copy(hiddenState, hiddenState + batchSize * numUnits, stateOutput); in executeTyped()
158 Shape hiddenState = context->getInputShape(kHiddenStateTensor); in prepare() local
173 NN_RET_CHECK_EQ(getNumberOfDimensions(hiddenState), 2); in prepare()
179 NN_RET_CHECK_EQ(batchSize, getSizeOfDimension(hiddenState, 0)); in prepare()
180 NN_RET_CHECK_EQ(numUnits, getSizeOfDimension(hiddenState, 1)); in prepare()
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 4033 ANeuralNetworksOperandType hiddenState = {.type = inputOperandCode, in unidirectionlSequenceRNNTest() local
4053 {input, weights, recurrentWeights, bias, hiddenState, activation, timeMajor}, {output}); in unidirectionlSequenceRNNTest()
|