diff options
author | James Qin <jamesqin@google.com> | 2018-04-06 11:56:08 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-06 12:00:45 -0700 |
commit | 4f7943f7358fc69af62dc280c6f6ba549ebe2167 (patch) | |
tree | 19bdd3dddebeb7d26f9685328d0598bb58347bc0 /tensorflow/stream_executor/dnn.cc | |
parent | f15c117c4f4d51a6660bf14b6d6cf73c52692cfb (diff) |
Support RNN profiling in StreamExecutor for CUDA GPUs.
This change hasn't applied autotune on TF Cudnn kernels, only provides lower level support.
PiperOrigin-RevId: 191919566
Diffstat (limited to 'tensorflow/stream_executor/dnn.cc')
-rw-r--r-- | tensorflow/stream_executor/dnn.cc | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/stream_executor/dnn.cc b/tensorflow/stream_executor/dnn.cc index 44144a0613..0a3c4bcf50 100644 --- a/tensorflow/stream_executor/dnn.cc +++ b/tensorflow/stream_executor/dnn.cc @@ -28,6 +28,10 @@ bool DnnSupport::GetConvolveAlgorithms( return false; } +bool DnnSupport::GetRnnAlgorithms(std::vector<AlgorithmDesc>* out_algorithms) { + return false; +} + bool DnnSupport::GetConvolveBackwardDataAlgorithms( bool with_winograd_nonfused, int cc_major, int cc_minor, std::vector<AlgorithmDesc>* out_algorithms) { |