A class that implements a 2D convolution operation using PyTorch, inheriting from MLFunction. More...

#include <functions.h>

Inheritance diagram for TorchConvolute:

Public Member Functions
	TorchConvolute (float weights, int dims_)
	Constructor that initializes the convolution operation with weights and dimensions.

void	apply (const SelectivityVector &rows, std::vector< VectorPtr > &args, const TypePtr &type, exec::EvalCtx &context, VectorPtr &output) const override
	Applies the 2D convolution operation to the input array using PyTorch.

float *	getTensor () const override
	Returns the tensor associated with this function.

std::string	getFuncName ()
	Returns the name of the function.

Public Member Functions inherited from MLFunction
virtual	~MLFunction ()=default
	Virtual destructor.

virtual std::vector< int >	getDims ()
	Returns the dimensions of the function.

virtual int	getNumDims ()
	Returns the number of dimensions of the function.

virtual CostEstimate	getCost (std::vector< int > inputDims)
	Estimates the computational cost of applying the function.

Static Public Member Functions
static std::vector< std::shared_ptr< exec::FunctionSignature > >	signatures ()
	Returns the function signatures supported by this class.

static std::string	getName ()
	Static method to return the name of the function.

Additional Inherited Members
Protected Member Functions inherited from MLFunction
double	getWeightedCost (std::string name, float cost)
	Calculates the weighted cost of the function.

std::vector< double >	getCoefficientVector (std::string name)
	Retrieves the cost coefficients for the function.

Protected Attributes inherited from MLFunction
std::vector< int >	dims
	Dimensions of the function.

Detailed Description

A class that implements a 2D convolution operation using PyTorch, inheriting from MLFunction.

This class provides functionality to apply a 2D convolution operation to an input array using PyTorch.

Constructor & Destructor Documentation

◆ TorchConvolute()

TorchConvolute::TorchConvolute	(	float *	weights,
		int *	dims_ )

inline

Constructor that initializes the convolution operation with weights and dimensions.

Parameters

weights	A pointer to the weight matrix for the convolution.
dims_	An array containing the dimensions of the convolution operation.

Member Function Documentation

◆ apply()

void TorchConvolute::apply	(	const SelectivityVector &	rows,
		std::vector< VectorPtr > &	args,
		const TypePtr &	type,
		exec::EvalCtx &	context,
		VectorPtr &	output ) const

inlineoverride

Applies the 2D convolution operation to the input array using PyTorch.

This method processes the input array, applies the convolution, and stores the result in the output vector.

Parameters

rows	A SelectivityVector specifying the rows to process.
args	A vector of input arguments (e.g., the input array).
type	The type of the output vector.
context	The execution context.
output	The output vector where the result will be stored.

◆ getFuncName()

std::string TorchConvolute::getFuncName ( )

inlinevirtual

Returns the name of the function.

Returns: The name of the function as a string.

Reimplemented from MLFunction.

◆ getName()

static std::string TorchConvolute::getName ( )

inlinestatic

Static method to return the name of the function.

Returns: The name of the function as a string ("torchconv2d").

◆ getTensor()

float * TorchConvolute::getTensor ( ) const

inlineoverridevirtual

Returns the tensor associated with this function.

Returns: A pointer to the weight matrix for the convolution.

Implements MLFunction.

◆ signatures()

static std::vector< std::shared_ptr< exec::FunctionSignature > > TorchConvolute::signatures ( )

inlinestatic

Returns the function signatures supported by this class.

Returns: A vector of shared pointers to FunctionSignature objects.

The documentation for this class was generated from the following file:

functions.h

Public Member Functions

Static Public Member Functions

Additional Inherited Members