Python setSpatialTransformerDescriptor Examples

Programming Language: Python

Namespace/Package Name: cupy.cuda.cudnn

Method/Function: setSpatialTransformerDescriptor

Examples at hotexamples.com: 2

Python setSpatialTransformerDescriptor - 2 examples found. These are the top rated real world Python examples of cupy.cuda.cudnn.setSpatialTransformerDescriptor extracted from open source projects. You can rate examples to help us improve the quality of examples.

Example #1

Show file

def create_spatial_transformer_descriptor(sampler_type, dtype, nb_dims, dim_A):
    desc = Descriptor(cudnn.createSpatialTransformerDescriptor(),
                      cudnn.destroySpatialTransformerDescriptor)
    data_type = get_data_type(dtype)

    cudnn.setSpatialTransformerDescriptor(desc.value, sampler_type, data_type,
                                          nb_dims, dim_A)
    return desc

Example #2

Show file

File: cudnn.py Project: hitottiez/chainer

def create_spatial_transformer_descriptor(sampler_type, dtype, nb_dims, dim_A):
    desc = Descriptor(cudnn.createSpatialTransformerDescriptor(),
                      cudnn.destroySpatialTransformerDescriptor)
    data_type = get_data_type(dtype)

    cudnn.setSpatialTransformerDescriptor(
        desc.value, sampler_type, data_type, nb_dims, dim_A)
    return desc