infini::ops::linalg::Lstsq Class Reference

#include <linalg_lstsq.h>

Inheritance diagram for infini::ops::linalg::Lstsq:

Public Member Functions
	Lstsq (const Tensor input, const Tensor b, const std::optional< double > rcond, const std::optional< std::string > driver, Tensor solution, Tensor residuals, Tensor rank, Tensor singular_values)
virtual void	operator() (const Tensor input, const Tensor b, const std::optional< double > rcond, const std::optional< std::string > driver, Tensor solution, Tensor residuals, Tensor rank, Tensor singular_values) const =0
Public Member Functions inherited from infini::ops::Operator< Lstsq >
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	b_shape_
Tensor::Strides	b_strides_
DataType	b_type_
Tensor::Shape	solution_shape_
Tensor::Strides	solution_strides_
DataType	solution_type_
Tensor::Shape	residuals_shape_
Tensor::Strides	residuals_strides_
DataType	residuals_type_
Tensor::Shape	rank_shape_
Tensor::Strides	rank_strides_
DataType	rank_type_
Tensor::Shape	singular_values_shape_
Tensor::Strides	singular_values_strides_
DataType	singular_values_type_
std::optional< double >	rcond_ {}
std::optional< std::string >	driver_ {}
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< Lstsq >
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< Lstsq >
static constexpr Device::Type	device_type_
static constexpr std::size_t	implementation_index_

Constructor & Destructor Documentation

Lstsq()

infini::ops::linalg::Lstsq::Lstsq ( const Tensor  input, const Tensor  b, const std::optional< double >  rcond, const std::optional< std::string >  driver, Tensor  solution, Tensor  residuals, Tensor  rank, Tensor  singular_values  )

inline

Member Function Documentation

operator()()

virtual void infini::ops::linalg::Lstsq::operator() ( const Tensor  input, const Tensor  b, const std::optional< double >  rcond, const std::optional< std::string >  driver, Tensor  solution, Tensor  residuals, Tensor  rank, Tensor  singular_values  ) const

pure virtual

Member Data Documentation

b_shape_

Tensor::Shape infini::ops::linalg::Lstsq::b_shape_

protected

b_strides_

Tensor::Strides infini::ops::linalg::Lstsq::b_strides_

protected

b_type_

DataType infini::ops::linalg::Lstsq::b_type_

protected

device_index_

int infini::ops::linalg::Lstsq::device_index_ {0}

protected

driver_

std::optional<std::string> infini::ops::linalg::Lstsq::driver_ {}

protected

input_shape_

Tensor::Shape infini::ops::linalg::Lstsq::input_shape_

protected

input_strides_

Tensor::Strides infini::ops::linalg::Lstsq::input_strides_

protected

input_type_

DataType infini::ops::linalg::Lstsq::input_type_

protected

rank_shape_

Tensor::Shape infini::ops::linalg::Lstsq::rank_shape_

protected

rank_strides_

Tensor::Strides infini::ops::linalg::Lstsq::rank_strides_

protected

rank_type_

DataType infini::ops::linalg::Lstsq::rank_type_

protected

rcond_

std::optional<double> infini::ops::linalg::Lstsq::rcond_ {}

protected

residuals_shape_

Tensor::Shape infini::ops::linalg::Lstsq::residuals_shape_

protected

residuals_strides_

Tensor::Strides infini::ops::linalg::Lstsq::residuals_strides_

protected

residuals_type_

DataType infini::ops::linalg::Lstsq::residuals_type_

protected

singular_values_shape_

Tensor::Shape infini::ops::linalg::Lstsq::singular_values_shape_

protected

singular_values_strides_

Tensor::Strides infini::ops::linalg::Lstsq::singular_values_strides_

protected

singular_values_type_

DataType infini::ops::linalg::Lstsq::singular_values_type_

protected

solution_shape_

Tensor::Shape infini::ops::linalg::Lstsq::solution_shape_

protected

solution_strides_

Tensor::Strides infini::ops::linalg::Lstsq::solution_strides_

protected

solution_type_

DataType infini::ops::linalg::Lstsq::solution_type_

protected

---

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

• linalg_lstsq.h