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

#include <max_pool3d_with_indices_backward.h>

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

Public Member Functions

 MaxPool3dWithIndicesBackward (const Tensor grad_output, const Tensor input, const Tensor indices, const std::vector< int64_t > kernel_size, const std::vector< int64_t > stride, const std::vector< int64_t > padding, const std::vector< int64_t > dilation, const bool ceil_mode, Tensor grad_input)
 
virtual void operator() (const Tensor grad_output, const Tensor input, const Tensor indices, const std::vector< int64_t > kernel_size, const std::vector< int64_t > stride, const std::vector< int64_t > padding, const std::vector< int64_t > dilation, const bool ceil_mode, Tensor grad_input) const =0
 
- Public Member Functions inherited from infini::ops::Operator< MaxPool3dWithIndicesBackward >
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 grad_output_shape_
 
Tensor::Strides grad_output_strides_
 
DataType grad_output_type_
 
Tensor::Shape input_shape_
 
Tensor::Strides input_strides_
 
DataType input_type_
 
Tensor::Shape indices_shape_
 
Tensor::Strides indices_strides_
 
DataType indices_type_
 
Tensor::Shape grad_input_shape_
 
Tensor::Strides grad_input_strides_
 
DataType grad_input_type_
 
std::vector< int64_t > kernel_size_ {}
 
std::vector< int64_t > stride_ {}
 
std::vector< int64_t > padding_ {}
 
std::vector< int64_t > dilation_ {}
 
bool ceil_mode_ {}
 
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< MaxPool3dWithIndicesBackward >
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< MaxPool3dWithIndicesBackward >
static constexpr Device::Type device_type_
 
static constexpr std::size_t implementation_index_
 

Constructor & Destructor Documentation

◆ MaxPool3dWithIndicesBackward()

infini::ops::MaxPool3dWithIndicesBackward::MaxPool3dWithIndicesBackward ( const Tensor  grad_output,
const Tensor  input,
const Tensor  indices,
const std::vector< int64_t >  kernel_size,
const std::vector< int64_t >  stride,
const std::vector< int64_t >  padding,
const std::vector< int64_t >  dilation,
const bool  ceil_mode,
Tensor  grad_input 
)
inline

Member Function Documentation

◆ operator()()

virtual void infini::ops::MaxPool3dWithIndicesBackward::operator() ( const Tensor  grad_output,
const Tensor  input,
const Tensor  indices,
const std::vector< int64_t >  kernel_size,
const std::vector< int64_t >  stride,
const std::vector< int64_t >  padding,
const std::vector< int64_t >  dilation,
const bool  ceil_mode,
Tensor  grad_input 
) const
pure virtual

Member Data Documentation

◆ ceil_mode_

bool infini::ops::MaxPool3dWithIndicesBackward::ceil_mode_ {}
protected

◆ device_index_

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

◆ dilation_

std::vector<int64_t> infini::ops::MaxPool3dWithIndicesBackward::dilation_ {}
protected

◆ grad_input_shape_

Tensor::Shape infini::ops::MaxPool3dWithIndicesBackward::grad_input_shape_
protected

◆ grad_input_strides_

Tensor::Strides infini::ops::MaxPool3dWithIndicesBackward::grad_input_strides_
protected

◆ grad_input_type_

DataType infini::ops::MaxPool3dWithIndicesBackward::grad_input_type_
protected

◆ grad_output_shape_

Tensor::Shape infini::ops::MaxPool3dWithIndicesBackward::grad_output_shape_
protected

◆ grad_output_strides_

Tensor::Strides infini::ops::MaxPool3dWithIndicesBackward::grad_output_strides_
protected

◆ grad_output_type_

DataType infini::ops::MaxPool3dWithIndicesBackward::grad_output_type_
protected

◆ indices_shape_

Tensor::Shape infini::ops::MaxPool3dWithIndicesBackward::indices_shape_
protected

◆ indices_strides_

Tensor::Strides infini::ops::MaxPool3dWithIndicesBackward::indices_strides_
protected

◆ indices_type_

DataType infini::ops::MaxPool3dWithIndicesBackward::indices_type_
protected

◆ input_shape_

Tensor::Shape infini::ops::MaxPool3dWithIndicesBackward::input_shape_
protected

◆ input_strides_

Tensor::Strides infini::ops::MaxPool3dWithIndicesBackward::input_strides_
protected

◆ input_type_

DataType infini::ops::MaxPool3dWithIndicesBackward::input_type_
protected

◆ kernel_size_

std::vector<int64_t> infini::ops::MaxPool3dWithIndicesBackward::kernel_size_ {}
protected

◆ padding_

std::vector<int64_t> infini::ops::MaxPool3dWithIndicesBackward::padding_ {}
protected

◆ stride_

std::vector<int64_t> infini::ops::MaxPool3dWithIndicesBackward::stride_ {}
protected
The documentation for this class was generated from the following file: