Searched refs:CreateRnnWorkspace (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 1485 port::StatusOr<DeviceMemory<uint8>> CreateRnnWorkspace( in CreateRnnWorkspace() function
1534 CreateRnnWorkspace(stream, cudnn, rnn_desc, input_desc, in DoRnnForwardImpl()
1694 CreateRnnWorkspace(stream, cudnn, rnn_desc, input_desc, in DoRnnBackwardImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 1967 bool CreateRnnWorkspace(Stream* stream, miopenHandle_t miopen_handle, in CreateRnnWorkspace() function
2040 if (!CreateRnnWorkspace(stream, miopen.handle(), rnn_desc, input_desc, in DoRnnForwardImpl()
2161 if (!CreateRnnWorkspace(stream, miopen.handle(), rnn_desc, input_desc, in DoRnnBackwardImpl()
|