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
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