Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DBidirectionalSequenceRNN.cpp83 Shape fwWeightsShape = context->getInputShape(kFwWeightsTensor); in executeTyped() local
175 const uint32_t fwNumUnits = getSizeOfDimension(fwWeightsShape, 0); in executeTyped()
197 fixedTimeAuxInputShape, fwHiddenState, fwBias, fwWeights, fwWeightsShape, in executeTyped()