Searched refs:fwRecurrentWeights (Results 1 – 3 of 3) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | BidirectionalSequenceRNN.cpp | 129 const T* fwRecurrentWeights = context->getInputBuffer<T>(kFwRecurrentWeightsTensor); in executeTyped() local
269 fwAuxWeights, fwAuxWeightsShape, fwRecurrentWeights, in executeTyped()
370 Shape fwRecurrentWeights = context->getInputShape(kFwRecurrentWeightsTensor); in prepare() local
396 NN_RET_CHECK_EQ(getNumberOfDimensions(fwRecurrentWeights), 2); in prepare()
406 NN_RET_CHECK_EQ(fwNumUnits, getSizeOfDimension(fwRecurrentWeights, 0)); in prepare()
407 NN_RET_CHECK_EQ(fwNumUnits, getSizeOfDimension(fwRecurrentWeights, 1)); in prepare()
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 3949 ANeuralNetworksOperandType fwRecurrentWeights = {.type = inputOperandCode, in bidirectionlSequenceRNNTest() local
3954 ANeuralNetworksOperandType bwRecurrentWeights = fwRecurrentWeights; in bidirectionlSequenceRNNTest()
3986 {input, fwWeights, fwRecurrentWeights, fwBias, fwHiddenState, in bidirectionlSequenceRNNTest()
|
| /frameworks/ml/nn/tools/api/ |
| D | types.spec | 3498 * * 2: fwRecurrentWeights.
|