Home
last modified time | relevance | path

Searched defs:recurrentWeightsShape (Results 1 – 2 of 2) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DRNN.cpp118 const T* recurrentWeightsData, const Shape& recurrentWeightsShape, in RNNStep()
140 const Shape& recurrentWeightsShape, const int32_t activation, in RNNStep()
DUnidirectionalSequenceRNN.cpp59 Shape recurrentWeightsShape = context->getInputShape(kRecurrentWeightsTensor); in executeTyped() local