nabu.processing.transpose module¶

class nabu.processing.transpose.TransposeBase(shape, dtype, dst_dtype=None, **backend_options)[source]¶

Bases: object

A class for transposing (out-of-place) a cuda or opencl array

KernelCls¶: alias of BaseClassError

ProcessingCls¶: alias of BaseClassError

dtype_to_ctype¶: alias of BaseClassError

backend = 'none'¶

class nabu.processing.transpose.CudaTranspose(shape, dtype, dst_dtype=None, **backend_options)[source]¶

Bases: TransposeBase

KernelCls¶: alias of CudaKernel

ProcessingCls¶: alias of CudaProcessing

dtype_to_ctype(dtype)¶

backend = 'cuda'¶

class nabu.processing.transpose.OpenCLTranspose(shape, dtype, dst_dtype=None, **backend_options)[source]¶

Bases: TransposeBase

KernelCls¶: alias of OpenCLKernel

ProcessingCls¶: alias of OpenCLProcessing

dtype_to_ctype(dtype)¶

backend = 'opencl'¶

nabu.processing.transpose.transpose(array, dst=None, **backend_options)[source]¶