diff options
author | Xiaoqiang Zheng <zhengxq@google.com> | 2016-08-19 10:01:32 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-08-19 11:18:09 -0700 |
commit | 859e47fd8ebcdd7fb1411fd0090d0e95a801e7cb (patch) | |
tree | ae0c4e4e690d53b92720049fb75acc119e4ea5e0 /tensorflow/stream_executor/platform | |
parent | de8838042fb34e53a511f25b4613611fc368beeb (diff) |
Add stream-executor changes to enable Cudnn fused LSTM/RNN support.
Change: 130770287
Diffstat (limited to 'tensorflow/stream_executor/platform')
-rw-r--r-- | tensorflow/stream_executor/platform/port.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/stream_executor/platform/port.h b/tensorflow/stream_executor/platform/port.h index 3c1b40b016..6603df4878 100644 --- a/tensorflow/stream_executor/platform/port.h +++ b/tensorflow/stream_executor/platform/port.h @@ -55,5 +55,7 @@ using tensorflow::LINKER_INITIALIZED; #define SE_DISALLOW_COPY_AND_ASSIGN TF_DISALLOW_COPY_AND_ASSIGN #define SE_MUST_USE_RESULT TF_MUST_USE_RESULT +#define SE_PREDICT_TRUE TF_PREDICT_TRUE +#define SE_PREDICT_FALSE TF_PREDICT_FALSE #endif // TENSORFLOW_STREAM_EXECUTOR_PLATFORM_PORT_H_ |