Searched refs:SpatialTransformerDescriptor (Results 1 – 3 of 3) sorted by relevance
62 SpatialTransformerDescriptor& desc, in setSamplerDescriptor()90 SpatialTransformerDescriptor desc; in cudnn_affine_grid_generator_forward()113 SpatialTransformerDescriptor desc; in cudnn_affine_grid_generator_backward()
55 SpatialTransformerDescriptor& desc, in setSamplerDescriptor()103 SpatialTransformerDescriptor desc; // sampler descriptor in cudnn_grid_sampler_forward()157 SpatialTransformerDescriptor desc; // sampler descriptor in cudnn_grid_sampler_backward()
217 struct TORCH_CUDA_CPP_API SpatialTransformerDescriptor struct