Searched refs:fwBias (Results 1 – 3 of 3) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | BidirectionalSequenceRNN.cpp | 131 const T* fwBias = context->getInputBuffer<T>(kFwBiasTensor); in executeTyped() local
268 fixedTimeAuxInputShape, fwHiddenState, fwBias, fwWeights, fwWeightsShape, in executeTyped()
371 Shape fwBias = context->getInputShape(kFwBiasTensor); in prepare() local
397 NN_RET_CHECK_EQ(getNumberOfDimensions(fwBias), 1); in prepare()
405 NN_RET_CHECK_EQ(fwNumUnits, getSizeOfDimension(fwBias, 0)); in prepare()
|
| /frameworks/ml/nn/runtime/test/ |
| D | TestValidateOperations.cpp | 3955 ANeuralNetworksOperandType fwBias = {.type = inputOperandCode, in bidirectionlSequenceRNNTest() local
3960 ANeuralNetworksOperandType bwBias = fwBias; in bidirectionlSequenceRNNTest()
3986 {input, fwWeights, fwRecurrentWeights, fwBias, fwHiddenState, in bidirectionlSequenceRNNTest()
|
| /frameworks/ml/nn/tools/api/ |
| D | types.spec | 3500 * * 3: fwBias.
|