Searched refs:hiddenStateOutput (Results 1 – 2 of 2) sorted by relevance
146 T* hiddenStateOutput) { in RNNStep() argument217 if (hiddenStateOutput != nullptr) { in RNNStep()218 *hiddenStateOutput = output_ptr_batch[o]; in RNNStep()219 ++hiddenStateOutput; in RNNStep()242 _Float16* hiddenStateOutput);257 float* outputData, float* hiddenStateOutput);
62 T* outputData, T* hiddenStateOutput = nullptr);