Searched defs:fn_dropout_state (Results 1 – 2 of 2) sorted by relevance
455 … const Tensor& fn_dropout_state = c10::value_or_else(fn_dropout_state_opt, [] {return Tensor();}); in miopen_rnn() local564 const Tensor& fn_dropout_state, const Tensor& fn_reserve, in miopen_rnn_backward_input()679 const Tensor& fn_dropout_state, const Tensor& fn_reserve, const Tensor& fn_workspace in miopen_rnn_backward_weight()
1403 const Tensor& fn_dropout_state = in _cudnn_rnn() local1687 const Tensor& fn_dropout_state, in _cudnn_rnn_backward_input()1917 const Tensor& fn_dropout_state, in _cudnn_rnn_backward_weight()