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

added comment.

上级 3f5e0726
...@@ -267,6 +267,18 @@ class GpuConv(Op): ...@@ -267,6 +267,18 @@ class GpuConv(Op):
version=-1, version=-1,
verbose=0, verbose=0,
kshp=None): kshp=None):
"""
:param version: each version of c_code implement many kernel for the
convolution. By default we try to guess the best one.
You can force one version with this parameter. This
parameter is used by the tests.
:param verbose: for value of 1,2 and 3. Print more information during
the execution of the convolution. Mostly used for
optimization or debugging.
:param kshp: The size of the kernel. If provided, can genera
faster code. If the GpuConv op is automatically inserted,
we take its value automatically from the Conv op.
"""
self.border_mode = border_mode self.border_mode = border_mode
self.subsample = subsample self.subsample = subsample
if logical_img_hw is not None: if logical_img_hw is not None:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论