TensorOps.Transfer.ixx File Reference

Tensor transfer utilities � copy/dispatch helpers for tensor data movement. More...

#include <concepts>
#include <string>
#include <memory>
#include <span>
#include <type_traits>
#include <stdexcept>
import Compute.CpuMemoryResource;
import Compute.ExecutionContext;
import Compute.DeviceType;
import Dnn.TensorOps.Base;
import Serialization.Tensor;
import Compute.IExecutionContext;
import Compute.DeviceId;
import Compute.Device;
import Dnn.TensorDataTypeTraits;
import Dnn.TensorDataTypeMap;
import Dnn.TensorDataType;
import Dnn.Tensor;

Namespaces
namespace	Mila
	Mila main API namespace.
namespace	Mila::Dnn

Functions
template<TensorDataType TSrcDataType, typename TSrcMemoryResource, TensorDataType TDstDataType, typename TDstMemoryResource> requires isValidTensor<TSrcDataType, TSrcMemoryResource> && isValidTensor<TDstDataType, TDstMemoryResource>
void	Mila::Dnn::copy (const Tensor< TSrcDataType, TSrcMemoryResource > &src, Tensor< TDstDataType, TDstMemoryResource > &dst, IExecutionContext *exec_context=nullptr)
	Copies tensor data from source to destination tensor with optional ExecutionContext.
template<TensorDataType TDstDataType, typename TDstMemoryResource> requires isValidTensor<TDstDataType, TDstMemoryResource>
void	Mila::Dnn::copyFromBlob (const Serialization::ITensorBlob &blob, Tensor< TDstDataType, TDstMemoryResource > &dst, IExecutionContext *exec_context=nullptr)
template<TensorDataType TSrcDataType, TensorDataType TDstDataType, typename TDstMemoryResource> requires isValidTensor<TDstDataType, TDstMemoryResource>
void	Mila::Dnn::copyFromBlobWithConversion (const Serialization::ITensorBlob &blob, Tensor< TDstDataType, TDstMemoryResource > &dst, IExecutionContext *exec_context=nullptr)
	Copy a serialized blob into a destination tensor, converting element types.
template<TensorDataType TDstDataType, TensorDataType TSrcDataType, typename TSrcMemoryResource> requires isValidTensor<TSrcDataType, TSrcMemoryResource> && isValidTensor<TDstDataType, CpuMemoryResource>
Tensor< TDstDataType, CpuMemoryResource >	Mila::Dnn::toHost (const Tensor< TSrcDataType, TSrcMemoryResource > &src, IExecutionContext *exec_context=nullptr)
	Create a host (CPU) tensor from src and copy data into it.

Detailed Description

Tensor transfer utilities � copy/dispatch helpers for tensor data movement.

Provides an exported generic copy() template that validates shapes, handles host/device accessibility, and dispatches to device-specific TensorOps implementations. Supports an optional execution context for stream control.