Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/cpu_operations/
DBidirectionalSequenceRNN.cpp112 const T* bwAuxWeights = nullptr; in executeTyped() local
118 bwAuxWeights = context->getInputBuffer<T>(kBwAuxWeightsTensor); in executeTyped()
266 bwAuxWeights, bwAuxWeightsShape, bwRecurrentWeights, in executeTyped()
320 Shape bwAuxWeights = context->getInputShape(kBwAuxWeightsTensor); in prepare() local
363 NN_RET_CHECK_EQ(getNumberOfDimensions(bwAuxWeights), 2u); in prepare()
369 NN_RET_CHECK_EQ(getSizeOfDimension(bwAuxWeights, 0), bwNumUnits); in prepare()
370 NN_RET_CHECK_EQ(getSizeOfDimension(bwAuxWeights, 1), getSizeOfDimension(auxInput, 2)); in prepare()
/packages/modules/NeuralNetworks/tools/api/
Dtypes.spec3687 * * 11:bwAuxWeights.
3695 * * 11:bwAuxWeights.