Dnn.Tensor Module Reference

Classes
class	Mila::Dnn::Tensor< TElementType, TMemoryResource >

Typedefs
template<class T >
using	Mila::Dnn::DeviceTensor = Tensor< T, Compute::CudaMemoryResource >
	Tensor type that uses device (GPU) memory.
template<typename T >
using	Mila::Dnn::HostTensor = Tensor< T, Compute::HostMemoryResource >
	Tensor type that uses host (CPU) memory.
template<class T >
using	Mila::Dnn::PinnedTensor = Tensor< T, Compute::CudaPinnedMemoryResource >
	Tensor type that uses pinned (page-locked) host memory.
template<class T >
using	Mila::Dnn::UniversalTensor = Tensor< T, Compute::CudaManagedMemoryResource >
	Tensor type that uses CUDA managed memory accessible from both CPU and GPU.

Files
file	/home/runner/work/Mila/Mila/Mila/Src/Dnn/Tensor/Tensor.ixx