InfiniOps
Operator Library for Accelerators
Loading...
Searching...
No Matches
infini::ops::internal::linalg::Slogdet Class Referenceabstract

#include <internal_linalg_slogdet.h>

Inheritance diagram for infini::ops::internal::linalg::Slogdet:
infini::ops::Operator< Slogdet > infini::ops::OperatorBase infini::ops::OperatorBase

Public Member Functions

 Slogdet (const Tensor A, Tensor sign, Tensor logabsdet, Tensor LU, Tensor pivots)
 
virtual void operator() (const Tensor A, Tensor sign, Tensor logabsdet, Tensor LU, Tensor pivots) const =0
 
- Public Member Functions inherited from infini::ops::Operator< Slogdet >
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 sign_shape_
 
Tensor::Strides sign_strides_
 
DataType sign_type_
 
Tensor::Shape logabsdet_shape_
 
Tensor::Strides logabsdet_strides_
 
DataType logabsdet_type_
 
Tensor::Shape LU_shape_
 
Tensor::Strides LU_strides_
 
DataType LU_type_
 
Tensor::Shape pivots_shape_
 
Tensor::Strides pivots_strides_
 
DataType pivots_type_
 
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< Slogdet >
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< Slogdet >
static constexpr Device::Type device_type_
 
static constexpr std::size_t implementation_index_
 

Constructor & Destructor Documentation

◆ Slogdet()

infini::ops::internal::linalg::Slogdet::Slogdet ( const Tensor  A,
Tensor  sign,
Tensor  logabsdet,
Tensor  LU,
Tensor  pivots 
)
inline

Member Function Documentation

◆ operator()()

virtual void infini::ops::internal::linalg::Slogdet::operator() ( const Tensor  A,
Tensor  sign,
Tensor  logabsdet,
Tensor  LU,
Tensor  pivots 
) const
pure virtual

Member Data Documentation

◆ A_shape_

Tensor::Shape infini::ops::internal::linalg::Slogdet::A_shape_
protected

◆ A_strides_

Tensor::Strides infini::ops::internal::linalg::Slogdet::A_strides_
protected

◆ A_type_

DataType infini::ops::internal::linalg::Slogdet::A_type_
protected

◆ device_index_

int infini::ops::internal::linalg::Slogdet::device_index_ {0}
protected

◆ logabsdet_shape_

Tensor::Shape infini::ops::internal::linalg::Slogdet::logabsdet_shape_
protected

◆ logabsdet_strides_

Tensor::Strides infini::ops::internal::linalg::Slogdet::logabsdet_strides_
protected

◆ logabsdet_type_

DataType infini::ops::internal::linalg::Slogdet::logabsdet_type_
protected

◆ LU_shape_

Tensor::Shape infini::ops::internal::linalg::Slogdet::LU_shape_
protected

◆ LU_strides_

Tensor::Strides infini::ops::internal::linalg::Slogdet::LU_strides_
protected

◆ LU_type_

DataType infini::ops::internal::linalg::Slogdet::LU_type_
protected

◆ pivots_shape_

Tensor::Shape infini::ops::internal::linalg::Slogdet::pivots_shape_
protected

◆ pivots_strides_

Tensor::Strides infini::ops::internal::linalg::Slogdet::pivots_strides_
protected

◆ pivots_type_

DataType infini::ops::internal::linalg::Slogdet::pivots_type_
protected

◆ sign_shape_

Tensor::Shape infini::ops::internal::linalg::Slogdet::sign_shape_
protected

◆ sign_strides_

Tensor::Strides infini::ops::internal::linalg::Slogdet::sign_strides_
protected

◆ sign_type_

DataType infini::ops::internal::linalg::Slogdet::sign_type_
protected

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