Searched refs:CudnnRnnParameters (Results 1 – 1 of 1) sorted by relevance
148 class CudnnRnnParameters { class150 CudnnRnnParameters(int num_layers, int input_size, int num_units, in CudnnRnnParameters() function in tensorflow::__anonebbbcdba0111::CudnnRnnParameters172 bool operator==(const CudnnRnnParameters& other) const { in operator ==()176 bool operator!=(const CudnnRnnParameters& other) const { in operator !=()226 AutoTuneSingleton<RnnAutoTuneGroup, CudnnRnnParameters, AlgorithmConfig>;1468 CudnnRnnParameters rnn_params( in MaybeAutoTune()