Searched refs:hiddenState (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | UnidirectionalSequenceRNN.cpp | 61 const T* hiddenState = context->getInputBuffer<T>(kHiddenStateTensor); in executeTyped() local
96 RNN::RNNStep<T>(input, fixedTimeInputShape, hiddenState, bias, weights, weightsShape, in executeTyped()
99 hiddenState = output; in executeTyped()
111 std::copy(hiddenState, hiddenState + batchSize * numUnits, stateOutput); in executeTyped()
123 Shape hiddenState = context->getInputShape(kHiddenStateTensor); in prepare() local
138 NN_RET_CHECK_EQ(getNumberOfDimensions(hiddenState), 2u); in prepare()
144 NN_RET_CHECK_EQ(batchSize, getSizeOfDimension(hiddenState, 0)); in prepare()
145 NN_RET_CHECK_EQ(numUnits, getSizeOfDimension(hiddenState, 1)); in prepare()
|
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | TestValidateOperations.cpp | 4072 ANeuralNetworksOperandType hiddenState = {.type = inputOperandCode, in unidirectionlSequenceRNNTest() local
4092 {input, weights, recurrentWeights, bias, hiddenState, activation, timeMajor}, {output}); in unidirectionlSequenceRNNTest()
|