experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #29174 from ROCmSoftwarePlatform:google_upstream_cudnn_rnn_ops

Browse Source 
PiperOrigin-RevId: 252146125
This commit is contained in:
TensorFlower Gardener 2019-06-07 17:10:31 -07:00
commit dfacd7618a
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/core/kernels/cudnn_rnn_ops.cc
View File

@ -46,10 +46,10 @@ limitations under the License.
#include "tensorflow/core/util/env_var.h"
#include "tensorflow/core/util/use_cudnn.h"
#if GOOGLE_CUDA
#if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
#include "tensorflow/core/platform/stream_executor.h"
#include "tensorflow/core/util/stream_executor_util.h"
#endif // GOOGLE_CUDA
#endif // GOOGLE_CUDA || TENSORFLOW_USE_ROCM
/*
* This module implements ops that fuse a multi-layer multi-step RNN/LSTM model
@ -77,7 +77,7 @@ namespace tensorflow {
using CPUDevice = Eigen::ThreadPoolDevice;
#if GOOGLE_CUDA
#if GOOGLE_CUDA || TENSORFLOW_USE_ROCM
using GPUDevice = Eigen::GpuDevice;
using se::Stream;
@ -1902,6 +1902,6 @@ TF_CALL_double(REGISTER_GPU);
// TODO(zhengxq): Add the conversion of Cudnn RNN Params from and to
// its canonical form.
#endif // GOOGLE_CUDA
#endif // GOOGLE_CUDA || TENSORFLOW_USE_ROCM
} // namespace tensorflow