Searched refs:bwAuxWeights (Results 1 – 1 of 1) sorted by relevance
98 const T* bwAuxWeights = nullptr; in executeTyped() local103 bwAuxWeights = context->getInputBuffer<T>(kBwAuxWeightsTensor); in executeTyped()227 bwAuxWeights, bwAuxWeightsShape, bwRecurrentWeights, in executeTyped()303 Shape bwAuxWeights = context->getInputShape(kBwAuxWeightsTensor); in prepare() local350 NN_RET_CHECK_EQ(getNumberOfDimensions(bwAuxWeights), 2); in prepare()356 NN_RET_CHECK_EQ(getSizeOfDimension(bwAuxWeights, 0), bwNumUnits); in prepare()357 NN_RET_CHECK_EQ(getSizeOfDimension(bwAuxWeights, 1), getSizeOfDimension(auxInput, 2)); in prepare()