Searched refs:rnn_algo (Results 1 – 1 of 1) sorted by relevance
1275 cudnnRNNAlgo_t rnn_algo = ToCudnnRNNAlgo(algorithm_config.algorithm()); in Create() local1314 /*rnnDesc=*/rnn_desc.get(), /*algo=*/rnn_algo, /*cellMode=*/rnn_mode, in Create()1329 /*direction=*/direction_mode, /*mode=*/rnn_mode, /*algo=*/rnn_algo, in Create()1350 if (rnn_algo == CUDNN_RNN_ALGO_PERSIST_DYNAMIC) { in Create()