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

Upgrade cudnn doc. Make it more readable and more linkable.

上级 8ac02419
......@@ -145,42 +145,67 @@ get an error when cuDNN can not be used with them, use this flag:
cudnn.h must be readable by everybody.
Functions
=========
* :ref:`Convolution <libdoc_cuda_dnn_convolution>`
* :ref:`Pooling <libdoc_cuda_dnn_pooling>`
* :ref:`Batch Normalization <libdoc_cuda_dnn_bn>`
* :ref:`Softmax <libdoc_cuda_dnn_softmax>`
* :ref:`Internal Ops <libdoc_cuda_dnn_internal_ops>`
.. _libdoc_cuda_dnn_convolution:
Convolution
===========
.. automodule:: theano.sandbox.cuda.dnn
:noindex:
:members: dnn_conv, dnn_pool, dnn_conv3d, dnn_gradweight, dnn_gradinput, dnn_pool, dnn_batch_normalization_train, dnn_batch_normalization_test
:members: dnn_conv, dnn_conv3d, dnn_gradweight, dnn_gradinput
Convolution Ops
===============
.. _libdoc_cuda_dnn_pooling:
Pooling
=======
.. automodule:: theano.sandbox.cuda.dnn
:noindex:
:members: GpuDnnConvDesc, GpuDnnConv, GpuDnnConv3d, GpuDnnConvGradW,
GpuDnnConv3dGradW, GpuDnnConvGradI, GpuDnnConv3dGradI
:members: dnn_pool
Pooling Ops
===========
.. _libdoc_cuda_dnn_bn:
Batch Normalization
===================
.. automodule:: theano.sandbox.cuda.dnn
:noindex:
:members: GpuDnnPoolDesc, GpuDnnPool, GpuDnnPoolGrad
:members: dnn_batch_normalization_train, dnn_batch_normalization_test
Softmax Ops
===========
.. _libdoc_cuda_dnn_rnn:
.. automodule:: theano.sandbox.cuda.dnn
RNN
===
New back-end only! Without dropout support.
.. automodule:: theano.gpuarray.dnn
:noindex:
:members: GpuDnnSoftmax, GpuDnnSoftmaxGrad
:members: RNNBlock
.. _libdoc_cuda_dnn_softmax:
Softmax
=======
.. _libdoc_cuda_dnn_bn:
You can manually use the op :ref:`GpuDnnSoftmax
<theano.sandbox.cuda.dnn.GpuDnnSoftmax>` to use its extra feature.
Batch Normalization
===================
.. _libdoc_cuda_dnn_internal_ops:
Internal Ops
============
.. automodule:: theano.sandbox.cuda.dnn
:noindex:
:members: dnn_batch_normalization_train, dnn_batch_normalization_test
:members: GpuDnnConvDesc, GpuDnnConv, GpuDnnConv3d, GpuDnnConvGradW,
GpuDnnConv3dGradW, GpuDnnConvGradI, GpuDnnConv3dGradI,
GpuDnnPoolDesc, GpuDnnPool, GpuDnnPoolGrad,
GpuDnnBatchNormInference, GpuDnnBatchNorm, GpuDnnBatchNormGrad,
GpuDnnSoftmax, GpuDnnSoftmaxGrad,
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论