numpydoc for theano/sandbox/gpuarray/opt.py

theano/sandbox/gpuarray/opt.py

@@ -89,7 +89,9 @@ def safe_to_cpu(x):
 def op_lifter(OP, cuda_only=False):
     """
     OP(..., host_from_gpu(), ...) -> host_from_gpu(GpuOP(...))
+    
     gpu_from_host(OP(inp0, ...)) -> GpuOP(inp0, ...)
+
     """
     def f(maker):
         def local_opt(node):
@@ -122,7 +124,10 @@ def op_lifter(OP, cuda_only=False):
 
 
 class InputToGpuOptimizer(Optimizer):
-    "Transfer the input to the gpu to start the rolling wave."
+    """
+    Transfer the input to the gpu to start the rolling wave.
+
+    """
 
     def add_requirements(self, fgraph):
         fgraph.attach_feature(toolbox.ReplaceValidate())
@@ -173,6 +178,7 @@ def local_gpuaalloc2(node):
     Join(axis, {Alloc or HostFromGPU}, ...) -> Join(axis, GpuAlloc, Alloc, ...)
 
     Moves an alloc that is an input to join to the gpu.
+
     """
     if (isinstance(node.op, tensor.Alloc) and
         all(c != 'output' and
@@ -654,6 +660,7 @@ def local_gpu_conv(node):
     gpu_from_host(conv) -> gpu_conv(gpu_from_host)
 
     conv(host_from_gpu) -> host_from_gpu(gpu_conv)
+
     """
     def GpuConvOp_from_ConvOp(op):
         logical_img_hw = None
@@ -698,7 +705,8 @@ def local_gpu_conv(node):
         return ret
 
     def values_eq_approx(a, b):
-        """This fct is needed to don't have DebugMode raise useless
+        """
+        This fct is needed to don't have DebugMode raise useless
         error due to ronding error.
 
         This happen as We reduce on the two last dimensions, so this
@@ -736,7 +744,10 @@ register_opt()(conv_groupopt)
 @register_opt("low_memory")
 @local_optimizer([GpuCAReduceCuda])
 def local_gpu_elemwise_careduce(node):
-    """ Merge some GpuCAReduceCuda and GPUElemwise"""
+    """
+    Merge some GpuCAReduceCuda and GPUElemwise.
+
+    """
     if (isinstance(node.op, GpuCAReduceCuda) and
             node.op.pre_scalar_op is None and
             node.inputs[0].owner and
@@ -767,10 +778,11 @@ def tensor_to_gpu(x):
 def gpu_safe_new(x, tag=''):
     """
     Internal function that constructs a new variable from x with the same
-    type, but with a different name ( old name + tag). This function is used
+    type, but with a different name (old name + tag). This function is used
     by gradient, or the R-op to construct new variables for the inputs of
     the inner graph such that there is no interference between the original
     graph and the newly constructed graph.
+
     """
     if hasattr(x, 'name') and x.name is not None:
         nw_name = x.name + tag
@@ -788,8 +800,9 @@ def gpu_reconstruct_graph(inputs, outputs, tag=None):
     """
     Different interface to clone, that allows you to pass inputs.
     Compared to clone, this method always replaces the inputs with
-    new variables of the same type, and returns those ( in the same
+    new variables of the same type, and returns those (in the same
     order as the original inputs).
+
     """
     if tag is None:
         tag = ''