Loading...
Searching...
No Matches
opennn::CudnnRnnState Member List

This is the complete list of members for opennn::CudnnRnnState, including all inherited members.

backend_stateopennn::CudnnRnnStatemutableprotected
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) constopennn::CudnnRnnStateprotected
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) constopennn::CudnnRnnStateprotected
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) constopennn::CudnnRnnStateprotected
cudnn_rnn_configuration_(const CudnnRnnShapeSlot &, const char *mode) constopennn::CudnnRnnStateprotected
cudnn_rnn_forward_(const CudnnRnnShapeSlot &, bool is_training, bool has_cell_state, const void *x, void *y, Buffer &forward_state, const function< CudnnRnnShapeSlot &()> &reconfigure) constopennn::CudnnRnnStateprotected
cudnn_setup_(const CudnnRnnConfig &, Index input_features, Index output_features, Index time_steps, Index batch_size, bool for_training) constopennn::CudnnRnnStateprotected
cudnn_setup_attempt_(const CudnnRnnConfig &, Index input_features, Index output_features, Index time_steps, Index batch_size, bool for_training) constopennn::CudnnRnnStateprotected
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) constopennn::CudnnRnnStateprotected
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) constopennn::CudnnRnnStateprotected
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) constopennn::CudnnRnnStateprotected
ensure_cudnn_setup_(Index batch_size, bool for_training) const =0opennn::CudnnRnnStateprotectedpure virtual
lock_backend_state() constopennn::CudnnRnnStateinlineprotected
pack_weights_to_cudnn_(Buffer &, uint64_t parameters_version) const =0opennn::CudnnRnnStateprotectedpure virtual
prepare_cudnn_forward_state_(Buffer &, bool is_training, const CudnnRnnShapeSlot &) constopennn::CudnnRnnStateprotected
unpack_gradients_from_cudnn_(Buffer &) const =0opennn::CudnnRnnStateprotectedpure virtual