infini::ops::linalg::LuFactorEx Class Reference

#include <linalg_lu_factor_ex.h>

Inheritance diagram for infini::ops::linalg::LuFactorEx:

Public Member Functions
	LuFactorEx (const Tensor A, const bool pivot, const bool check_errors, Tensor LU, Tensor pivots, Tensor info)
virtual void	operator() (const Tensor A, const bool pivot, const bool check_errors, Tensor LU, Tensor pivots, Tensor info) const =0
Public Member Functions inherited from infini::ops::Operator< LuFactorEx >
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	A_shape_
Tensor::Strides	A_strides_
DataType	A_type_
Tensor::Shape	LU_shape_
Tensor::Strides	LU_strides_
DataType	LU_type_
Tensor::Shape	pivots_shape_
Tensor::Strides	pivots_strides_
DataType	pivots_type_
Tensor::Shape	info_shape_
Tensor::Strides	info_strides_
DataType	info_type_
bool	pivot_ {}
bool	check_errors_ {}
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< LuFactorEx >
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< LuFactorEx >
static constexpr Device::Type	device_type_
static constexpr std::size_t	implementation_index_

Constructor & Destructor Documentation

LuFactorEx()

infini::ops::linalg::LuFactorEx::LuFactorEx ( const Tensor  A, const bool  pivot, const bool  check_errors, Tensor  LU, Tensor  pivots, Tensor  info  )

inline

Member Function Documentation

operator()()

virtual void infini::ops::linalg::LuFactorEx::operator() ( const Tensor  A, const bool  pivot, const bool  check_errors, Tensor  LU, Tensor  pivots, Tensor  info  ) const

pure virtual

Member Data Documentation

A_shape_

Tensor::Shape infini::ops::linalg::LuFactorEx::A_shape_

protected

A_strides_

Tensor::Strides infini::ops::linalg::LuFactorEx::A_strides_

protected

A_type_

DataType infini::ops::linalg::LuFactorEx::A_type_

protected

check_errors_

bool infini::ops::linalg::LuFactorEx::check_errors_ {}

protected

device_index_

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

protected

info_shape_

Tensor::Shape infini::ops::linalg::LuFactorEx::info_shape_

protected

info_strides_

Tensor::Strides infini::ops::linalg::LuFactorEx::info_strides_

protected

info_type_

DataType infini::ops::linalg::LuFactorEx::info_type_

protected

LU_shape_

Tensor::Shape infini::ops::linalg::LuFactorEx::LU_shape_

protected

LU_strides_

Tensor::Strides infini::ops::linalg::LuFactorEx::LU_strides_

protected

LU_type_

DataType infini::ops::linalg::LuFactorEx::LU_type_

protected

pivot_

bool infini::ops::linalg::LuFactorEx::pivot_ {}

protected

pivots_shape_

Tensor::Shape infini::ops::linalg::LuFactorEx::pivots_shape_

protected

pivots_strides_

Tensor::Strides infini::ops::linalg::LuFactorEx::pivots_strides_

protected

pivots_type_

DataType infini::ops::linalg::LuFactorEx::pivots_type_

protected

---

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

• linalg_lu_factor_ex.h