Searched refs:fwAuxWeights (Results 1 – 1 of 1) sorted by relevance
97 const T* fwAuxWeights = nullptr; in executeTyped() local102 fwAuxWeights = context->getInputBuffer<T>(kFwAuxWeightsTensor); in executeTyped()198 fwAuxWeights, fwAuxWeightsShape, fwRecurrentWeights, in executeTyped()302 Shape fwAuxWeights = context->getInputShape(kFwAuxWeightsTensor); in prepare() local349 NN_RET_CHECK_EQ(getNumberOfDimensions(fwAuxWeights), 2); in prepare()354 NN_RET_CHECK_EQ(getSizeOfDimension(fwAuxWeights, 0), fwNumUnits); in prepare()355 NN_RET_CHECK_EQ(getSizeOfDimension(fwAuxWeights, 1), getSizeOfDimension(auxInput, 2)); in prepare()