Grangeat-based 2D/3D image registration
Loading...
Searching...
No Matches
Public Member Functions | List of all members
reg23::CUDATexture2D Class Reference

#include <CUDATexture.h>

Public Member Functions

 CUDATexture2D (const at::Tensor &tensor, const std::string &addressModeX, const std::string &addressModeY)
 
 CUDATexture2D (const at::Tensor &tensor, Vec< TextureAddressMode, 2 > addressModes)
 
unsigned long long Handle () const
 
at::Tensor SizeTensor () const
 
void CleanUp () noexcept
 Cleans up the underlying PyTorch tensor, CUDA texture and CUDA array.
 

Constructor & Destructor Documentation

◆ CUDATexture2D() [1/2]

reg23::CUDATexture2D::CUDATexture2D ( const at::Tensor &  tensor,
const std::string &  addressModeX,
const std::string &  addressModeY 
)

◆ CUDATexture2D() [2/2]

reg23::CUDATexture2D::CUDATexture2D ( const at::Tensor &  tensor,
Vec< TextureAddressMode, 2 >  addressModes 
)

Member Function Documentation

◆ CleanUp()

void reg23::CUDATexture2D::CleanUp ( )
noexcept

Cleans up the underlying PyTorch tensor, CUDA texture and CUDA array.

This is effectively the destructor, but it is defined separately so it can be called independently. This is because it doesn't seem to be possible to have the destructor of this class called automatically on destruction when used via Python bindings.

◆ Handle()

unsigned long long reg23::CUDATexture2D::Handle ( ) const

◆ SizeTensor()

at::Tensor reg23::CUDATexture2D::SizeTensor ( ) const

The documentation for this class was generated from the following file: