Searched refs:recurrentWeights (Results 1 – 2 of 2) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | UnidirectionalSequenceRNN.cpp | 58 const T* recurrentWeights = context->getInputBuffer<T>(kRecurrentWeightsTensor); in executeTyped() local
97 recurrentWeights, recurrentWeightsShape, activation, output); in executeTyped()
121 Shape recurrentWeights = context->getInputShape(kRecurrentWeightsTensor); in prepare() local
136 NN_RET_CHECK_EQ(getNumberOfDimensions(recurrentWeights), 2u); in prepare()
142 NN_RET_CHECK_EQ(numUnits, getSizeOfDimension(recurrentWeights, 0)); in prepare()
143 NN_RET_CHECK_EQ(numUnits, getSizeOfDimension(recurrentWeights, 1)); in prepare()
|
| /packages/modules/NeuralNetworks/runtime/test/ |
| D | TestValidateOperations.cpp | 3059 ANeuralNetworksOperandType recurrentWeights = floatTensor2D; in TEST() local
3068 {input, weights, recurrentWeights, bias, hiddenStateIn, activation}, in TEST()
3094 ANeuralNetworksOperandType recurrentWeights = floatTensor2D; in TEST() local
3103 {input, weights, recurrentWeights, bias, hiddenStateIn, activation}, in TEST()
4062 ANeuralNetworksOperandType recurrentWeights = {.type = inputOperandCode, in unidirectionlSequenceRNNTest() local
4092 {input, weights, recurrentWeights, bias, hiddenState, activation, timeMajor}, {output}); in unidirectionlSequenceRNNTest()
|