提交 e521f6fa authored 作者: Frederic Bastien's avatar Frederic Bastien

Do the same cudnn doc fix to the gpuarray back-end and move the RNN doc to it.

上级 8292c229
...@@ -43,7 +43,7 @@ To get an error if Theano can not use cuDNN, use this Theano flag: ...@@ -43,7 +43,7 @@ To get an error if Theano can not use cuDNN, use this Theano flag:
.. note:: .. note::
cuDNN v5rc is supported in Theano master version. So it dropped cuDNN v3 support. cuDNN v5.1 is supported in Theano master version. So it dropped cuDNN v3 support.
Theano 0.8.0 and 0.8.1 support only cuDNN v3 and v4. Theano 0.8.0 and 0.8.1 support only cuDNN v3 and v4.
Theano 0.8.2 will support only v4 and v5. Theano 0.8.2 will support only v4 and v5.
...@@ -132,30 +132,70 @@ To get an error if Theano can not use cuDNN, use this Theano flag: ...@@ -132,30 +132,70 @@ To get an error if Theano can not use cuDNN, use this Theano flag:
cudnn.h must be readable by everybody. cudnn.h must be readable by everybody.
Functions * :ref:`Convolution <libdoc_gpuarray_dnn_convolution>`
========= * :ref:`Pooling <libdoc_gpuarray_dnn_pooling>`
* :ref:`Batch Normalization <libdoc_gpuarray_dnn_bn>`
* :ref:`RNN <libdoc_gpuarray_dnn_rnn>`
* :ref:`Softmax <libdoc_gpuarray_dnn_softmax>`
* :ref:`Internal Ops <libdoc_gpuarray_dnn_internal_ops>`
dnn_present, dnn_available
.. _libdoc_gpuarray_dnn_convolution:
Convolution
===========
.. automodule:: theano.gpuarray.dnn .. automodule:: theano.gpuarray.dnn
:noindex: :noindex:
:members: dnn_conv, dnn_pool :members: dnn_conv, dnn_conv3d, dnn_gradweight, dnn_gradweight3d, dnn_gradinput, dnn_gradinput3d
.. _libdoc_gpuarray_dnn_pooling:
Convolution Ops Pooling
=============== =======
.. automodule:: theano.gpuarray.dnn .. automodule:: theano.gpuarray.dnn
:noindex: :noindex:
:members: GpuDnnConvDesc, GpuDnnConv, GpuDnnConvGradW, GpuDnnConvGradI :members: dnn_pool
Pooling Ops .. _libdoc_gpuarray_dnn_bn:
===========
Batch Normalization
===================
.. automodule:: theano.gpuarray.dnn .. automodule:: theano.gpuarray.dnn
:noindex: :noindex:
:members: GpuDnnPoolDesc, GpuDnnPool, GpuDnnPoolGrad :members: dnn_batch_normalization_train, dnn_batch_normalization_test
Softmax Ops .. _libdoc_gpuarray_dnn_rnn:
===========
RNN
===
New back-end only! Without dropout support.
.. automodule:: theano.gpuarray.dnn .. automodule:: theano.gpuarray.dnn
:noindex: :noindex:
:members: GpuDnnSoftmax, GpuDnnSoftmaxGrad :members: RNNBlock
.. _libdoc_gpuarray_dnn_softmax:
Softmax
=======
You can manually use the op :class:`GpuDnnSoftmax
<theano.gpuarray.dnn.GpuDnnSoftmax>` to use its extra feature.
.. _libdoc_gpuarray_dnn_internal_ops:
Internal Ops
============
.. automodule:: theano.gpuarray.dnn
:noindex:
:members: GpuDnnConvDesc, GpuDnnConv, GpuDnnConvGradW, GpuDnnConvGradI,
GpuDnnPoolDesc, GpuDnnPool, GpuDnnPoolGrad,
GpuDnnBatchNormInference, GpuDnnBatchNorm, GpuDnnBatchNormGrad,
GpuDnnSoftmax, GpuDnnSoftmaxGrad
...@@ -9,7 +9,7 @@ ...@@ -9,7 +9,7 @@
`cuDNN <https://developer.nvidia.com/cuDNN>`_ is an NVIDIA library with `cuDNN <https://developer.nvidia.com/cuDNN>`_ is an NVIDIA library with
functionality used by deep neural network. It provides optimized versions functionality used by deep neural network. It provides optimized versions
of some operations like the convolution. cuDNN is not currently of some operations like the convolution. cuDNN is not currently
installed with CUDA 6.5. You must download and install it installed with CUDA. You must download and install it
yourself. yourself.
To install it, decompress the downloaded file and make the ``*.h`` and To install it, decompress the downloaded file and make the ``*.h`` and
...@@ -46,7 +46,7 @@ get an error when cuDNN can not be used with them, use this flag: ...@@ -46,7 +46,7 @@ get an error when cuDNN can not be used with them, use this flag:
.. note:: .. note::
cuDNN v5rc is supported in Theano master version. So it dropped cuDNN v3 support. cuDNN v5.1 is supported in Theano master version. So it dropped cuDNN v3 support.
Theano 0.8.0 and 0.8.1 support only cuDNN v3 and v4. Theano 0.8.0 and 0.8.1 support only cuDNN v3 and v4.
Theano 0.8.2 will support only v4 and v5. Theano 0.8.2 will support only v4 and v5.
...@@ -184,11 +184,8 @@ Batch Normalization ...@@ -184,11 +184,8 @@ Batch Normalization
RNN RNN
=== ===
New back-end only! Without dropout support. `New back-end only! <libdoc_gpuarray_dnn_rnn>`_
.. automodule:: theano.gpuarray.dnn
:noindex:
:members: RNNBlock
.. _libdoc_cuda_dnn_softmax: .. _libdoc_cuda_dnn_softmax:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论