Searched refs:hiddenStateInputData (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | RNN.cpp | 120 bool RNN::RNNStep(const T* inputData, const Shape& inputShape, const T* hiddenStateInputData, in RNNStep() argument
129 hiddenStateInputData, biasData, weightsData, weightsShape, in RNNStep()
141 const Shape& auxInputShape, const T* hiddenStateInputData, const T* biasData, in RNNStep() argument
167 const T* hidden_state_in_ptr_batch = hiddenStateInputData + b * num_units; in RNNStep()
228 const _Float16* hiddenStateInputData, const _Float16* biasData,
235 const _Float16* hiddenStateInputData, const _Float16* biasData,
244 const float* hiddenStateInputData, const float* biasData,
251 const float* hiddenStateInputData, const float* biasData,
|
| D | RNN.h | 50 static bool RNNStep(const T* inputData, const Shape& inputShape, const T* hiddenStateInputData,
57 const Shape& auxInputShape, const T* hiddenStateInputData,
|