Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DBidirectionalSequenceRNN.cpp111 const T* fwAuxWeights = nullptr; in executeTyped() local
117 fwAuxWeights = context->getInputBuffer<T>(kFwAuxWeightsTensor); in executeTyped()
238 fwAuxWeights, fwAuxWeightsShape, fwRecurrentWeights, in executeTyped()
319 Shape fwAuxWeights = context->getInputShape(kFwAuxWeightsTensor); in prepare() local
362 NN_RET_CHECK_EQ(getNumberOfDimensions(fwAuxWeights), 2u); in prepare()
367 NN_RET_CHECK_EQ(getSizeOfDimension(fwAuxWeights, 0), fwNumUnits); in prepare()
368 NN_RET_CHECK_EQ(getSizeOfDimension(fwAuxWeights, 1), getSizeOfDimension(auxInput, 2)); in prepare()
/packages/modules/NeuralNetworks/tools/api/
Dtypes.spec3684 * * 10:fwAuxWeights.
3693 * * 10:fwAuxWeights.