TensorPtr.ixx File Reference

Memory-type-aware pointer wrappers for tensor data access. More...

#include <stdexcept>

Classes
class	Mila::Dnn::TensorPtr< T, IsHostAccessible >
	Base tensor pointer class that wraps a raw pointer with memory-type safety. More...

Namespaces
namespace	Mila
namespace	Mila::Dnn

Typedefs
template<typename T >
using	Mila::Dnn::DevicePtr = TensorPtr< T, false >
	Type alias for device memory pointers.
template<typename T >
using	Mila::Dnn::HostPtr = TensorPtr< T, true >
	Type alias for host memory pointers.

Functions
template<typename T , bool IsHostAccessible>
T *	Mila::Dnn::raw_pointer_cast (const TensorPtr< T, IsHostAccessible > &ptr) noexcept
	Gets a raw pointer from a TensorPtr (similar to thrust::raw_pointer_cast)

Detailed Description

Memory-type-aware pointer wrappers for tensor data access.