Searched refs:bwBias (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | BidirectionalSequenceRNN.cpp | 93 const T* bwBias = context->getInputBuffer<T>(kBwBiasTensor); in executeTyped() local
226 fixedTimeAuxInputShape, bwHiddenState, bwBias, bwWeights, bwWeightsShape, in executeTyped()
298 Shape bwBias = context->getInputShape(kBwBiasTensor); in prepare() local
330 NN_RET_CHECK_EQ(getNumberOfDimensions(bwBias), 1); in prepare()
341 NN_RET_CHECK_EQ(bwNumUnits, getSizeOfDimension(bwBias, 0)); in prepare()
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 2961 ANeuralNetworksOperandType bwBias = fwBias; in bidirectionlSequenceRNNTest() local
2988 bwWeights, bwRecurrentWeights, bwBias, bwHiddenState, input, in bidirectionlSequenceRNNTest()
|