Searched refs:bwRecurrentWeights (Results 1 – 3 of 3) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | BidirectionalSequenceRNN.cpp | 136 const T* bwRecurrentWeights = context->getInputBuffer<T>(kBwRecurrentWeightsTensor); in executeTyped() local
297 bwAuxWeights, bwAuxWeightsShape, bwRecurrentWeights, in executeTyped()
374 Shape bwRecurrentWeights = context->getInputShape(kBwRecurrentWeightsTensor); in prepare() local
400 NN_RET_CHECK_EQ(getNumberOfDimensions(bwRecurrentWeights), 2); in prepare()
415 NN_RET_CHECK_EQ(bwNumUnits, getSizeOfDimension(bwRecurrentWeights, 0)); in prepare()
416 NN_RET_CHECK_EQ(bwNumUnits, getSizeOfDimension(bwRecurrentWeights, 1)); in prepare()
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 3954 ANeuralNetworksOperandType bwRecurrentWeights = fwRecurrentWeights; in bidirectionlSequenceRNNTest() local
3987 bwWeights, bwRecurrentWeights, bwBias, bwHiddenState, input, in bidirectionlSequenceRNNTest()
|
| /frameworks/ml/nn/tools/api/ |
| D | types.spec | 3507 * * 6: bwRecurrentWeights.
|