Searched refs:recurrentWeightsShape (Results 1 – 3 of 3) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | RNN.cpp | 118 const T* recurrentWeightsData, const Shape& recurrentWeightsShape, in RNNStep() argument
127 recurrentWeightsData, recurrentWeightsShape, activation, in RNNStep()
140 const Shape& recurrentWeightsShape, const int32_t activation, in RNNStep() argument
149 const uint32_t recurrent_weights_stride = recurrentWeightsShape.dimensions[1]; in RNNStep()
227 const Shape& recurrentWeightsShape, int32_t activation,
235 const Shape& recurrentWeightsShape, const int32_t activation,
243 const Shape& recurrentWeightsShape, int32_t activation,
251 const Shape& recurrentWeightsShape, int32_t activation,
|
| D | UnidirectionalSequenceRNN.cpp | 59 Shape recurrentWeightsShape = context->getInputShape(kRecurrentWeightsTensor); in executeTyped() local
97 recurrentWeights, recurrentWeightsShape, activation, output); in executeTyped()
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | RNN.h | 58 const T* recurrentWeightsData, const Shape& recurrentWeightsShape,
66 const T* recurrentWeightsData, const Shape& recurrentWeightsShape,
|