Home
last modified time | relevance | path

Searched refs:CheckRNNParameterSize (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc1469 port::Status CheckRNNParameterSize( in CheckRNNParameterSize() function
1532 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()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc1951 bool CheckRNNParameterSize( in CheckRNNParameterSize() function
2033 if (!CheckRNNParameterSize(miopen.handle(), rnn_desc, input_desc)) { in DoRnnForwardImpl()
2154 if (!CheckRNNParameterSize(miopen.handle(), rnn_desc, input_desc)) { in DoRnnBackwardImpl()