Searched refs:hiddenStateOutput (Results 1 – 2 of 2) sorted by relevance
142 T* hiddenStateOutput) { in RNNStep() argument213 if (hiddenStateOutput != nullptr) { in RNNStep()214 *hiddenStateOutput = output_ptr_batch[o]; in RNNStep()215 ++hiddenStateOutput; in RNNStep()238 _Float16* hiddenStateOutput);253 float* outputData, float* hiddenStateOutput);
62 T* outputData, T* hiddenStateOutput = nullptr);