BinaryOperation.ixx File Reference

Abstract base class for binary operations in the Mila neural network framework. More...

#include <memory>
#include <vector>
#include <type_traits>
#include <stdexcept>
import Compute.CudaMemoryResource;
import Compute.DeviceContextHelpers;
import Compute.DeviceContext;
import Compute.CpuMemoryResource;
import Compute.OperationAttributes;
import Compute.OperationType;
import Compute.OperationBase;
import Compute.DeviceType;
import Compute.CudaDevice;
import Compute.ComputeDevice;
import Compute.Precision;
import Dnn.TensorTraits;
import Dnn.Tensor;

Classes
class	Mila::Dnn::Compute::BinaryOperation< TDeviceType, TInput1, TInput2, TOutput >
	Abstract class for binary operations in the neural network framework. More...

Namespaces
namespace	Mila
namespace	Mila::Dnn
namespace	Mila::Dnn::Compute

Detailed Description

Abstract base class for binary operations in the Mila neural network framework.

Provides an interface for operations that take two input tensors and produce a single output tensor. This serves as the foundation for operations like addition, multiplication, convolution, and other binary neural network functions. Supports configurable compute precision policies for mixed-precision computation.