InfiniOps
Operator Library for Accelerators
Loading...
Searching...
No Matches
Public Member Functions | Protected Attributes | List of all members
infini::ops::Searchsorted Class Referenceabstract

#include <searchsorted.h>

Inheritance diagram for infini::ops::Searchsorted:
infini::ops::Operator< Searchsorted > infini::ops::OperatorBase infini::ops::OperatorBase

Public Member Functions

 Searchsorted (const Tensor sorted_sequence, const Tensor input, const std::optional< Tensor > sorter, const bool out_int32, const bool right, const std::optional< std::string > side, Tensor out)
 
 Searchsorted (const Tensor sorted_sequence, const Tensor input, const bool out_int32, const bool right, const std::optional< std::string > side, const std::optional< Tensor > sorter, Tensor out)
 
 Searchsorted (const Tensor sorted_sequence, const double input, const std::optional< Tensor > sorter, const bool out_int32, const bool right, const std::optional< std::string > side, Tensor out)
 
 Searchsorted (const Tensor sorted_sequence, const double input, const bool out_int32, const bool right, const std::optional< std::string > side, const std::optional< Tensor > sorter, Tensor out)
 
void operator() (const Tensor sorted_sequence, const Tensor input, const std::optional< Tensor > sorter, const bool out_int32, const bool right, const std::optional< std::string > side, Tensor out) const
 
void operator() (const Tensor sorted_sequence, const double input, const std::optional< Tensor > sorter, const bool out_int32, const bool right, const std::optional< std::string > side, Tensor out) const
 
virtual void operator() (const Tensor sorted_sequence, const Tensor input, const bool out_int32, const bool right, const std::optional< std::string > side, const std::optional< Tensor > sorter, Tensor out) const =0
 
virtual void operator() (const Tensor sorted_sequence, const double input, const bool out_int32, const bool right, const std::optional< std::string > side, const std::optional< Tensor > sorter, Tensor out) const =0
 
- Public Member Functions inherited from infini::ops::Operator< Searchsorted >
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 sorted_sequence_shape_
 
Tensor::Strides sorted_sequence_strides_
 
DataType sorted_sequence_type_
 
Tensor::Shape input_shape_
 
Tensor::Strides input_strides_
 
DataType input_type_
 
Tensor::Shape out_shape_
 
Tensor::Strides out_strides_
 
DataType out_type_
 
bool has_sorter_ {false}
 
Tensor::Shape sorter_shape_
 
Tensor::Strides sorter_strides_
 
DataType sorter_type_ {DataType::kFloat32}
 
bool out_int32_ {}
 
bool right_ {}
 
std::optional< std::string > side_ {}
 
double input_ {}
 
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< Searchsorted >
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< Searchsorted >
static constexpr Device::Type device_type_
 
static constexpr std::size_t implementation_index_
 

Constructor & Destructor Documentation

◆ Searchsorted() [1/4]

infini::ops::Searchsorted::Searchsorted ( const Tensor  sorted_sequence,
const Tensor  input,
const std::optional< Tensor >  sorter,
const bool  out_int32,
const bool  right,
const std::optional< std::string >  side,
Tensor  out 
)
inline

◆ Searchsorted() [2/4]

infini::ops::Searchsorted::Searchsorted ( const Tensor  sorted_sequence,
const Tensor  input,
const bool  out_int32,
const bool  right,
const std::optional< std::string >  side,
const std::optional< Tensor >  sorter,
Tensor  out 
)
inline
Deprecated:
Use (sorted_sequence, input, sorter, out_int32, right, side, out). This constructor will be removed in a future release.

◆ Searchsorted() [3/4]

infini::ops::Searchsorted::Searchsorted ( const Tensor  sorted_sequence,
const double  input,
const std::optional< Tensor >  sorter,
const bool  out_int32,
const bool  right,
const std::optional< std::string >  side,
Tensor  out 
)
inline

◆ Searchsorted() [4/4]

infini::ops::Searchsorted::Searchsorted ( const Tensor  sorted_sequence,
const double  input,
const bool  out_int32,
const bool  right,
const std::optional< std::string >  side,
const std::optional< Tensor >  sorter,
Tensor  out 
)
inline
Deprecated:
Use (sorted_sequence, input, sorter, out_int32, right, side, out). This constructor will be removed in a future release.

Member Function Documentation

◆ operator()() [1/4]

virtual void infini::ops::Searchsorted::operator() ( const Tensor  sorted_sequence,
const double  input,
const bool  out_int32,
const bool  right,
const std::optional< std::string >  side,
const std::optional< Tensor >  sorter,
Tensor  out 
) const
pure virtual
Deprecated:
Place sorter before attributes. This overload will be removed in a future release.

◆ operator()() [2/4]

void infini::ops::Searchsorted::operator() ( const Tensor  sorted_sequence,
const double  input,
const std::optional< Tensor >  sorter,
const bool  out_int32,
const bool  right,
const std::optional< std::string >  side,
Tensor  out 
) const
inline

◆ operator()() [3/4]

virtual void infini::ops::Searchsorted::operator() ( const Tensor  sorted_sequence,
const Tensor  input,
const bool  out_int32,
const bool  right,
const std::optional< std::string >  side,
const std::optional< Tensor >  sorter,
Tensor  out 
) const
pure virtual
Deprecated:
Place sorter before attributes. This overload will be removed in a future release.

◆ operator()() [4/4]

void infini::ops::Searchsorted::operator() ( const Tensor  sorted_sequence,
const Tensor  input,
const std::optional< Tensor >  sorter,
const bool  out_int32,
const bool  right,
const std::optional< std::string >  side,
Tensor  out 
) const
inline

Member Data Documentation

◆ device_index_

int infini::ops::Searchsorted::device_index_ {0}
protected

◆ has_sorter_

bool infini::ops::Searchsorted::has_sorter_ {false}
protected

◆ input_

double infini::ops::Searchsorted::input_ {}
protected

◆ input_shape_

Tensor::Shape infini::ops::Searchsorted::input_shape_
protected

◆ input_strides_

Tensor::Strides infini::ops::Searchsorted::input_strides_
protected

◆ input_type_

DataType infini::ops::Searchsorted::input_type_
protected

◆ out_int32_

bool infini::ops::Searchsorted::out_int32_ {}
protected

◆ out_shape_

Tensor::Shape infini::ops::Searchsorted::out_shape_
protected

◆ out_strides_

Tensor::Strides infini::ops::Searchsorted::out_strides_
protected

◆ out_type_

DataType infini::ops::Searchsorted::out_type_
protected

◆ right_

bool infini::ops::Searchsorted::right_ {}
protected

◆ side_

std::optional<std::string> infini::ops::Searchsorted::side_ {}
protected

◆ sorted_sequence_shape_

Tensor::Shape infini::ops::Searchsorted::sorted_sequence_shape_
protected

◆ sorted_sequence_strides_

Tensor::Strides infini::ops::Searchsorted::sorted_sequence_strides_
protected

◆ sorted_sequence_type_

DataType infini::ops::Searchsorted::sorted_sequence_type_
protected

◆ sorter_shape_

Tensor::Shape infini::ops::Searchsorted::sorter_shape_
protected

◆ sorter_strides_

Tensor::Strides infini::ops::Searchsorted::sorter_strides_
protected

◆ sorter_type_

DataType infini::ops::Searchsorted::sorter_type_ {DataType::kFloat32}
protected
The documentation for this class was generated from the following file: