GridSample3D.h File Reference

Implementations of grid sampling functions. More...

#include "Common.h"
#include "Texture.h"

Go to the source code of this file.

Classes
struct	reg23::GridSample3D< texture_t >
struct	reg23::GridSample3D< texture_t >::CommonData

Namespaces
namespace	reg23

Functions
at::Tensor	reg23::GridSample3D_CPU (const at::Tensor &input, const at::Tensor &grid, const std::string &addressModeX, const std::string &addressModeY, const std::string &addressModeZ, c10::optional< at::Tensor > out)
	Sample the given 3D input tensor at the positions given in grid according to the given address mode using bilinear interpolation. This implementation is single-threaded.
__host__ at::Tensor	reg23::GridSample3D_CUDA (const at::Tensor &input, const at::Tensor &grid, const std::string &addressModeX, const std::string &addressModeY, const std::string &addressModeZ, c10::optional< at::Tensor > out)
	An implementation of reg23::GridSample3D_CPU that uses CUDA parallelisation.

Detailed Description

Implementations of grid sampling functions.