Expand description
RNN / LSTM / GRU forward + backward training requests.
Routes through cuDNN’s RNN v8 API (cudnnRNNForward /
cudnnRNNBackwardData_v8 / cudnnRNNBackwardWeights_v8).
Structs§
- RnnBwd
Request - RNN backward (data + weights) request.
- RnnFwd
Request - RNN forward request.
- RnnParams
- RNN parameters.
Enums§
- RnnDirection
- Direction.
- RnnMode
- RNN cell mode.
Functions§
- build_
rnn_ fwd_ spec - Build a spec-side
OperationGraphSpecfor 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.