Searched refs:_TestCudnnCompatibleRnnCellsHelper (Results 1 – 1 of 1) sorted by relevance
856 self._TestCudnnCompatibleRnnCellsHelper(CUDNN_LSTM)861 self._TestCudnnCompatibleRnnCellsHelper(CUDNN_GRU)866 self._TestCudnnCompatibleRnnCellsHelper(CUDNN_RNN_TANH)871 self._TestCudnnCompatibleRnnCellsHelper(CUDNN_RNN_RELU)873 def _TestCudnnCompatibleRnnCellsHelper(self, rnn_mode): member in CudnnRNNTestCompatibleRNNCells