提交 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: ...@@ -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. 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 .. automodule:: theano.sandbox.cuda.dnn
:noindex: :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 .. automodule:: theano.sandbox.cuda.dnn
:noindex: :noindex:
:members: GpuDnnConvDesc, GpuDnnConv, GpuDnnConv3d, GpuDnnConvGradW, :members: dnn_pool
GpuDnnConv3dGradW, GpuDnnConvGradI, GpuDnnConv3dGradI
Pooling Ops .. _libdoc_cuda_dnn_bn:
===========
Batch Normalization
===================
.. automodule:: theano.sandbox.cuda.dnn .. automodule:: theano.sandbox.cuda.dnn
:noindex: :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: :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 .. automodule:: theano.sandbox.cuda.dnn
:noindex: :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 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论