Searched defs:fn_reserve (Results 1 – 2 of 2) sorted by relevance
564 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()
1688 const Tensor& fn_reserve, in _cudnn_rnn_backward_input()1918 const Tensor& fn_reserve) { in _cudnn_rnn_backward_weight()