Searched refs:bwBias (Results 1 – 3 of 3) sorted by relevance
/frameworks/ml/nn/common/operations/ |
D | BidirectionalSequenceRNN.cpp | 138 const T* bwBias = context->getInputBuffer<T>(kBwBiasTensor); in executeTyped() local 296 fixedTimeAuxInputShape, bwHiddenState, bwBias, bwWeights, bwWeightsShape, in executeTyped() 375 Shape bwBias = context->getInputShape(kBwBiasTensor); in prepare() local 401 NN_RET_CHECK_EQ(getNumberOfDimensions(bwBias), 1); in prepare() 414 NN_RET_CHECK_EQ(bwNumUnits, getSizeOfDimension(bwBias, 0)); in prepare()
|
/frameworks/ml/nn/runtime/test/ |
D | TestValidateOperations.cpp | 3960 ANeuralNetworksOperandType bwBias = fwBias; in bidirectionlSequenceRNNTest() local 3987 bwWeights, bwRecurrentWeights, bwBias, bwHiddenState, input, in bidirectionlSequenceRNNTest()
|
/frameworks/ml/nn/tools/api/ |
D | types.spec | 3509 * * 7: bwBias.
|