TensorOps.Structural.ixx File Reference

Device-dispatched structural operations for tensors. More...

#include <type_traits>
import Compute.ExecutionContext;
import Dnn.TensorOps.Base;
import Dnn.TensorTypes;
import Dnn.TensorDataType;
import Compute.DeviceType;
import Dnn.TensorDataTypeTraits;
import Dnn.ITensor;
import Dnn.TensorDataTypeMap;
import Dnn.Tensor;

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

Functions
template<TensorDataType TDataType, typename TMemoryResource> requires isValidTensor<TDataType, TMemoryResource>
void	Mila::Dnn::split (const Tensor< TDataType, TMemoryResource > &input, Tensor< TDataType, TMemoryResource > &output_a, Tensor< TDataType, TMemoryResource > &output_b, IExecutionContext *exec_context=nullptr)
template<TensorDataType TDataType, typename TMemoryResource> requires isValidTensor<TDataType, TMemoryResource>
void	Mila::Dnn::split (const Tensor< TDataType, TMemoryResource > &input, Tensor< TDataType, TMemoryResource > &output_a, Tensor< TDataType, TMemoryResource > &output_b, Tensor< TDataType, TMemoryResource > &output_c, IExecutionContext *exec_context=nullptr)

Detailed Description

Device-dispatched structural operations for tensors.