Searched refs:CheckRNNParameterSize (Results 1 – 2 of 2) sorted by relevance
1469 port::Status CheckRNNParameterSize( in CheckRNNParameterSize() function1532 SE_RETURN_IF_ERROR(CheckRNNParameterSize(cudnn, rnn_desc, input_desc)); in DoRnnForwardImpl()1692 SE_RETURN_IF_ERROR(CheckRNNParameterSize(cudnn, rnn_desc, input_desc)); in DoRnnBackwardImpl()
1951 bool CheckRNNParameterSize( in CheckRNNParameterSize() function2033 if (!CheckRNNParameterSize(miopen.handle(), rnn_desc, input_desc)) { in DoRnnForwardImpl()2154 if (!CheckRNNParameterSize(miopen.handle(), rnn_desc, input_desc)) { in DoRnnBackwardImpl()