Home
last modified time | relevance | path

Searched refs:fwAuxWeights (Results 1 – 4 of 4) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DBidirectionalSequenceRNN.cpp140 const T* fwAuxWeights = nullptr; in executeTyped() local
146 fwAuxWeights = context->getInputBuffer<T>(kFwAuxWeightsTensor); in executeTyped()
267 fwAuxWeights, fwAuxWeightsShape, fwRecurrentWeights, in executeTyped()
378 Shape fwAuxWeights = context->getInputShape(kFwAuxWeightsTensor); in prepare() local
421 NN_RET_CHECK_EQ(getNumberOfDimensions(fwAuxWeights), 2); in prepare()
426 NN_RET_CHECK_EQ(getSizeOfDimension(fwAuxWeights, 0), fwNumUnits); in prepare()
427 NN_RET_CHECK_EQ(getSizeOfDimension(fwAuxWeights, 1), getSizeOfDimension(auxInput, 2)); in prepare()
/aosp12/packages/modules/NeuralNetworks/tools/api/
H A Dtypes.spec3627 * * 10:fwAuxWeights.
3636 * * 10:fwAuxWeights.
/aosp12/hardware/interfaces/neuralnetworks/1.2/
H A Dtypes.hal2637 * * 10:fwAuxWeights.
/aosp12/hardware/interfaces/neuralnetworks/1.3/
H A Dtypes.hal2784 * * 10:fwAuxWeights.