Searched refs:set_algo (Results 1 – 2 of 2) sorted by relevance
90 void set_algo(miopenRNNAlgo_t algo) { in set_algo() function494 fn.rnn.set_algo(algo); in miopen_rnn()628 fn.rnn.set_algo(algo); in miopen_rnn_backward_input()722 fn.rnn.set_algo(algo); in miopen_rnn_backward_weight()
194 void set_algo(cudnnRNNAlgo_t algo) { in set_algo() function1478 fn.rnn.set_algo(algo); in _cudnn_rnn()1805 fn.rnn.set_algo(algo); in _cudnn_rnn_backward_input()1993 fn.rnn.set_algo(algo); in _cudnn_rnn_backward_weight()