infini::ops::Sub Class Reference

#include <sub.h>

Inheritance diagram for infini::ops::Sub:

Public Member Functions
	Sub (const Tensor input, const Tensor other, const double alpha, Tensor out)
	Sub (Tensor input, const Tensor other, const double alpha)
	Sub (const Tensor input, const double other, const double alpha, Tensor out)
virtual void	operator() (const Tensor input, const Tensor other, const double alpha, Tensor out) const =0
virtual void	operator() (Tensor input, const Tensor other, const double alpha) const =0
virtual void	operator() (const Tensor input, const double other, const double alpha, Tensor out) const =0
Public Member Functions inherited from infini::ops::Operator< Sub >
void	operator() (const Handle &handle, const Args &... args)
void	operator() (const Args &... args) const
void	operator() (const Handle &handle, const Args &... args)
void	operator() (const Args &... args) const
Public Member Functions inherited from infini::ops::OperatorBase
virtual	~OperatorBase ()=default
virtual std::size_t	workspace_size_in_bytes () const
void	set_handle (const Handle &handle)
void	set_config (const Config &config)
void	set_stream (void *stream)
void	set_workspace (void *workspace)
void	set_workspace_size_in_bytes (std::size_t workspace_size_in_bytes)
virtual	~OperatorBase ()=default
virtual std::size_t	workspace_size_in_bytes () const
void	set_handle (const Handle &handle)
void	set_config (const Config &config)
void	set_stream (void *stream)
void	set_workspace (void *workspace)
void	set_workspace_size_in_bytes (std::size_t workspace_size_in_bytes)

Protected Attributes
Tensor::Shape	input_shape_
Tensor::Strides	input_strides_
DataType	input_type_
Tensor::Shape	other_shape_
Tensor::Strides	other_strides_
DataType	other_type_
Tensor::Shape	out_shape_
Tensor::Strides	out_strides_
DataType	out_type_
double	alpha_ {}
double	other_ {}
int	device_index_ {0}
Protected Attributes inherited from infini::ops::OperatorBase
std::unique_ptr< Handle >	handle_ptr_
std::unique_ptr< Config >	config_ptr_
void *	stream_ {nullptr}
void *	workspace_ {nullptr}
std::size_t	workspace_size_in_bytes_ {0}

Additional Inherited Members
Static Public Member Functions inherited from infini::ops::Operator< Sub >
static void	clear_cache ()
static void	clear_cache ()
static std::unique_ptr< Operator >	Make (const Config &config, const Tensor tensor, Args &&... args)
static std::unique_ptr< Operator >	Make (const Tensor tensor, Args &&... args)
static std::unique_ptr< Operator >	Make (const Config &config, const std::vector< Tensor > tensors, Args &&... args)
static std::unique_ptr< Operator >	Make (const std::vector< Tensor > tensors, Args &&... args)
static std::unique_ptr< Operator >	Make (const Config &config, const Tensor tensor, Args &&... args)
static std::unique_ptr< Operator >	Make (const Tensor tensor, Args &&... args)
static std::unique_ptr< Operator >	Make (const Config &config, const std::vector< Tensor > tensors, Args &&... args)
static std::unique_ptr< Operator >	Make (const std::vector< Tensor > tensors, Args &&... args)
static void	Call (const Handle &handle, const Config &config, const Args &... args)
static void	Call (const Tensor tensor, const Args &... args)
static auto	Call (const TensorLike &tensor, const Args &... args)
static void	Call (const Handle &handle, const Config &config, const Args &... args)
static void	Call (const Tensor tensor, const Args &... args)
static auto	Call (const TensorLike &tensor, const Args &... args)
static std::vector< std::size_t >	active_implementation_indices (Device::Type dev_type)
static std::vector< std::size_t >	active_implementation_indices (Device::Type dev_type)
Static Protected Attributes inherited from infini::ops::Operator< Sub >
static constexpr Device::Type	device_type_
static constexpr std::size_t	implementation_index_

Constructor & Destructor Documentation

Sub() [1/3]

infini::ops::Sub::Sub ( const Tensor  input, const Tensor  other, const double  alpha, Tensor  out  )

inline

Sub() [2/3]

infini::ops::Sub::Sub ( Tensor  input, const Tensor  other, const double  alpha  )

inline

Deprecated:

Use the explicit-output constructor. This constructor will be removed in a future release.

Sub() [3/3]

infini::ops::Sub::Sub ( const Tensor  input, const double  other, const double  alpha, Tensor  out  )

inline

Member Function Documentation

operator()() [1/3]

virtual void infini::ops::Sub::operator() ( const Tensor  input, const double  other, const double  alpha, Tensor  out  ) const

pure virtual

operator()() [2/3]

virtual void infini::ops::Sub::operator() ( const Tensor  input, const Tensor  other, const double  alpha, Tensor  out  ) const

pure virtual

operator()() [3/3]

virtual void infini::ops::Sub::operator() ( Tensor  input, const Tensor  other, const double  alpha  ) const

pure virtual

Deprecated:

Use the explicit-output overload. This overload will be removed in a future release.

Member Data Documentation

alpha_

double infini::ops::Sub::alpha_ {}

protected

device_index_

int infini::ops::Sub::device_index_ {0}

protected

input_shape_

Tensor::Shape infini::ops::Sub::input_shape_

protected

input_strides_

Tensor::Strides infini::ops::Sub::input_strides_

protected

input_type_

DataType infini::ops::Sub::input_type_

protected

other_

double infini::ops::Sub::other_ {}

protected

other_shape_

Tensor::Shape infini::ops::Sub::other_shape_

protected

other_strides_

Tensor::Strides infini::ops::Sub::other_strides_

protected

other_type_

DataType infini::ops::Sub::other_type_

protected

out_shape_

Tensor::Shape infini::ops::Sub::out_shape_

protected

out_strides_

Tensor::Strides infini::ops::Sub::out_strides_

protected

out_type_

DataType infini::ops::Sub::out_type_

protected

---

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

• sub.h