infini::ops::Scatter Class Reference

#include <scatter.h>

Inheritance diagram for infini::ops::Scatter:

Public Member Functions
	Scatter (const Tensor input, const Tensor index, const Tensor src, const int64_t dim, Tensor out)
	Scatter (const Tensor input, const Tensor index, const double value, const int64_t dim, Tensor out)
	Scatter (const Tensor input, const Tensor index, const Tensor src, const int64_t dim, const std::string reduce, Tensor out)
	Scatter (const Tensor input, const Tensor index, const double value, const int64_t dim, const std::string reduce, Tensor out)
virtual void	operator() (const Tensor input, const Tensor index, const Tensor src, const int64_t dim, Tensor out) const =0
virtual void	operator() (const Tensor input, const Tensor index, const double value, const int64_t dim, Tensor out) const =0
virtual void	operator() (const Tensor input, const Tensor index, const Tensor src, const int64_t dim, const std::string reduce, Tensor out) const =0
virtual void	operator() (const Tensor input, const Tensor index, const double value, const int64_t dim, const std::string reduce, Tensor out) const =0
Public Member Functions inherited from infini::ops::Operator< Scatter >
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	index_shape_
Tensor::Strides	index_strides_
DataType	index_type_
Tensor::Shape	src_shape_
Tensor::Strides	src_strides_
DataType	src_type_
Tensor::Shape	out_shape_
Tensor::Strides	out_strides_
DataType	out_type_
int64_t	dim_ {}
double	value_ {}
std::string	reduce_ {}
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< Scatter >
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< Scatter >
static constexpr Device::Type	device_type_
static constexpr std::size_t	implementation_index_

Constructor & Destructor Documentation

Scatter() [1/4]

infini::ops::Scatter::Scatter ( const Tensor  input, const Tensor  index, const Tensor  src, const int64_t  dim, Tensor  out  )

inline

Scatter() [2/4]

infini::ops::Scatter::Scatter ( const Tensor  input, const Tensor  index, const double  value, const int64_t  dim, Tensor  out  )

inline

Scatter() [3/4]

infini::ops::Scatter::Scatter ( const Tensor  input, const Tensor  index, const Tensor  src, const int64_t  dim, const std::string  reduce, Tensor  out  )

inline

Scatter() [4/4]

infini::ops::Scatter::Scatter ( const Tensor  input, const Tensor  index, const double  value, const int64_t  dim, const std::string  reduce, Tensor  out  )

inline

Member Function Documentation

operator()() [1/4]

virtual void infini::ops::Scatter::operator() ( const Tensor  input, const Tensor  index, const double  value, const int64_t  dim, const std::string  reduce, Tensor  out  ) const

pure virtual

operator()() [2/4]

virtual void infini::ops::Scatter::operator() ( const Tensor  input, const Tensor  index, const double  value, const int64_t  dim, Tensor  out  ) const

pure virtual

operator()() [3/4]

virtual void infini::ops::Scatter::operator() ( const Tensor  input, const Tensor  index, const Tensor  src, const int64_t  dim, const std::string  reduce, Tensor  out  ) const

pure virtual

operator()() [4/4]

virtual void infini::ops::Scatter::operator() ( const Tensor  input, const Tensor  index, const Tensor  src, const int64_t  dim, Tensor  out  ) const

pure virtual

Member Data Documentation

device_index_

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

protected

dim_

int64_t infini::ops::Scatter::dim_ {}

protected

index_shape_

Tensor::Shape infini::ops::Scatter::index_shape_

protected

index_strides_

Tensor::Strides infini::ops::Scatter::index_strides_

protected

index_type_

DataType infini::ops::Scatter::index_type_

protected

input_shape_

Tensor::Shape infini::ops::Scatter::input_shape_

protected

input_strides_

Tensor::Strides infini::ops::Scatter::input_strides_

protected

input_type_

DataType infini::ops::Scatter::input_type_

protected

out_shape_

Tensor::Shape infini::ops::Scatter::out_shape_

protected

out_strides_

Tensor::Strides infini::ops::Scatter::out_strides_

protected

out_type_

DataType infini::ops::Scatter::out_type_

protected

reduce_

std::string infini::ops::Scatter::reduce_ {}

protected

src_shape_

Tensor::Shape infini::ops::Scatter::src_shape_

protected

src_strides_

Tensor::Strides infini::ops::Scatter::src_strides_

protected

src_type_

DataType infini::ops::Scatter::src_type_

protected

value_

double infini::ops::Scatter::value_ {}

protected

---

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

• scatter.h