Skip to main content

Module rnn

Module rnn 

Source
Expand description

RNN / LSTM / GRU forward + backward training requests.

Routes through cuDNN’s RNN v8 API (cudnnRNNForward / cudnnRNNBackwardData_v8 / cudnnRNNBackwardWeights_v8).

Structs§

RnnBwdRequest
RNN backward (data + weights) request.
RnnFwdRequest
RNN forward request.
RnnParams
RNN parameters.

Enums§

RnnDirection
Direction.
RnnMode
RNN cell mode.

Functions§

build_rnn_fwd_spec
Build a spec-side OperationGraphSpec for plan-cache keying. The actual RNN API does not use the v9 backend graph descriptor builder — but we keep a stable signature surface so the cache key machinery is shared.