Dnn.TensorPtr Module Reference

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

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)

Files
file	/home/runner/work/Mila/Mila/Mila/Src/Dnn/Tensor/TensorPtr.ixx
	Memory-type-aware pointer wrappers for tensor data access.