Searched defs:dropout_state_opt (Results 1 – 2 of 2) sorted by relevance
43 … train, bool bidirectional, IntArrayRef batch_sizes, const std::optional<Tensor>& dropout_state_op… in miopen_rnn_backward()763 … train, bool bidirectional, IntArrayRef batch_sizes, const std::optional<Tensor>& dropout_state_op… in miopen_rnn_backward()
90 const std::optional<Tensor>& dropout_state_opt, in _cudnn_rnn_backward()2108 const std::optional<Tensor>& dropout_state_opt, in _cudnn_rnn_backward()