Searched refs:miopen (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 2029 auto miopen = miopen_->GetHandle(parent_, stream); in DoRnnForwardImpl() local
2033 if (!CheckRNNParameterSize(miopen.handle(), rnn_desc, input_desc)) { in DoRnnForwardImpl()
2040 if (!CreateRnnWorkspace(stream, miopen.handle(), rnn_desc, input_desc, in DoRnnForwardImpl()
2053 miopen.handle() /*handle*/, rnn_desc.handle() /*rnnDesc*/, in DoRnnForwardImpl()
2076 miopen.handle() /*handle*/, rnn_desc.handle() /*rnnDesc*/, in DoRnnForwardImpl()
2094 miopen.handle() /*handle*/, rnn_desc.handle() /*rnnDesc*/, in DoRnnForwardImpl()
2150 auto miopen = miopen_->GetHandle(parent_, stream); in DoRnnBackwardImpl() local
2154 if (!CheckRNNParameterSize(miopen.handle(), rnn_desc, input_desc)) { in DoRnnBackwardImpl()
2161 if (!CreateRnnWorkspace(stream, miopen.handle(), rnn_desc, input_desc, in DoRnnBackwardImpl()
2186 miopen.handle() /*handle*/, rnn_desc.handle() /*rnnDesc*/, in DoRnnBackwardImpl()
[all …]
|
| D | BUILD | 248 "@local_config_rocm//rocm:miopen",
|
| /external/tensorflow/tensorflow/tools/ci_build/ |
| D | Dockerfile.rocm | 61 rocfft miopen-hip miopengemm rocblas hipblas rocrand \
|