Loading...
Searching...
No Matches
opennn::CudnnRnnState Member List
This is the complete list of members for opennn::CudnnRnnState, including all inherited members.
| backend_state | opennn::CudnnRnnState | mutableprotected |
| cudnn_copy_weight_regions_(int num_linear_layers, Index input_features, Index output_features, const TensorView *const *matrices, const TensorView *const *vectors, Buffer &packed_weights, bool to_cudnn) const | opennn::CudnnRnnState | protected |
| cudnn_pack_weights_(int num_linear_layers, Index input_features, Index output_features, const TensorView *const *weights, const TensorView *const *biases, Buffer &forward_state, uint64_t parameters_version) const | opennn::CudnnRnnState | protected |
| cudnn_rnn_backward_(const CudnnRnnShapeSlot &, bool has_cell_state, const void *x, const void *y, const void *dy, void *dx, const Buffer &forward_state, Buffer &backward_scratch) const | opennn::CudnnRnnState | protected |
| cudnn_rnn_configuration_(const CudnnRnnShapeSlot &, const char *mode) const | opennn::CudnnRnnState | protected |
| cudnn_rnn_forward_(const CudnnRnnShapeSlot &, bool is_training, bool has_cell_state, const void *x, void *y, Buffer &forward_state, const function< CudnnRnnShapeSlot &()> &reconfigure) const | opennn::CudnnRnnState | protected |
| cudnn_setup_(const CudnnRnnConfig &, Index input_features, Index output_features, Index time_steps, Index batch_size, bool for_training) const | opennn::CudnnRnnState | protected |
| cudnn_setup_attempt_(const CudnnRnnConfig &, Index input_features, Index output_features, Index time_steps, Index batch_size, bool for_training) const | opennn::CudnnRnnState | protected |
| cudnn_unpack_gradients_(int num_linear_layers, Index input_features, Index output_features, const TensorView *const *weight_gradients, const TensorView *const *bias_gradients, Buffer &backward_scratch) const | opennn::CudnnRnnState | protected |
| drive_cudnn_backward_(const CudnnRnnDims &, const TensorView &input, const TensorView &sequence_output, const TensorView &output_delta, const TensorView &cudnn_input_sequence, const TensorView &cudnn_output_sequence, TensorView &input_delta, TensorView &sequence_delta_scratch, TensorView &input_delta_scratch, const Buffer &forward_state, Buffer &backward_scratch) const | opennn::CudnnRnnState | protected |
| drive_cudnn_forward_(const CudnnRnnDims &, const TensorView &input, TensorView &sequence_output, TensorView &output, TensorView &cudnn_input_sequence, TensorView &cudnn_output_sequence, Buffer &forward_state, bool is_training, uint64_t parameters_version) const | opennn::CudnnRnnState | protected |
| ensure_cudnn_setup_(Index batch_size, bool for_training) const =0 | opennn::CudnnRnnState | protectedpure virtual |
| lock_backend_state() const | opennn::CudnnRnnState | inlineprotected |
| pack_weights_to_cudnn_(Buffer &, uint64_t parameters_version) const =0 | opennn::CudnnRnnState | protectedpure virtual |
| prepare_cudnn_forward_state_(Buffer &, bool is_training, const CudnnRnnShapeSlot &) const | opennn::CudnnRnnState | protected |
| unpack_gradients_from_cudnn_(Buffer &) const =0 | opennn::CudnnRnnState | protectedpure virtual |