mir::ops::AvgPool2DOp Class Reference

#include <AvgPool2DOp.h>

Collaboration diagram for mir::ops::AvgPool2DOp:

Public Member Functions
	AvgPool2DOp (Output *arg, const AvgPool2DOpAttributes &attributes)
Operation *	copyWithInputs (const std::vector< Output * > &inputs) override
const std::vector< std::int32_t > &	getWindowSize () const
const std::vector< std::int32_t > &	getStrides () const
const std::vector< std::int32_t > &	getPaddingBefore () const
const std::vector< std::int32_t > &	getPaddingAfter () const
bool	getIncludePad () const
DataFormat	getDataFormat () const
const AvgPool2DOpAttributes &	getAttributes () const
Public Member Functions inherited from mir::Operation
virtual	~Operation ()=default
Type	getType () const
std::size_t	getId () const
void	setId (std::size_t id)
std::size_t	getNumInputs () const
std::size_t	getNumOutputs () const
std::deque< Output * > &	getInputs ()
const std::deque< Output * > &	getInputs () const
std::deque< Output > &	getOutputs ()
const std::deque< Output > &	getOutputs () const
Output *	getInput (std::size_t index)
const Output *	getInput (std::size_t index) const
Output *	getOutput (std::size_t index)
const Output *	getOutput (std::size_t index) const
const Shape &	getInputShape (std::size_t index) const
const Shape &	getOutputShape (std::size_t index) const
void	accept (IVisitor *v)

Additional Inherited Members
Public Types inherited from mir::Operation
enum class	Type { HANDLE_OP }
Protected Member Functions inherited from mir::Operation
	Operation (Type type, const std::vector< Output * > &inputs, std::size_t num_outputs=1)
void	setOutputType (std::size_t index, const TensorType &type)

Detailed Description

Definition at line 31 of file AvgPool2DOp.h.

Constructor & Destructor Documentation

AvgPool2DOp()

mir::ops::AvgPool2DOp::AvgPool2DOp ( Output *  arg, const AvgPool2DOpAttributes &  attributes  )

inline

Definition at line 34 of file AvgPool2DOp.h.

    : Operation(Type::avgPool2D, {arg}), _attributes(attributes)
  {
    inferOutputTypes();
  }

Member Function Documentation

copyWithInputs()

Operation * mir::ops::AvgPool2DOp::copyWithInputs ( const std::vector< Output * > &  inputs )

inlineoverridevirtual

Implements mir::Operation.

Definition at line 40 of file AvgPool2DOp.h.

  {
    return new AvgPool2DOp(inputs[0], _attributes);
  };

getAttributes()

const AvgPool2DOpAttributes & mir::ops::AvgPool2DOp::getAttributes ( ) const

inline

Definition at line 57 of file AvgPool2DOp.h.

{ return _attributes; }

getDataFormat()

DataFormat mir::ops::AvgPool2DOp::getDataFormat ( ) const

inline

Definition at line 55 of file AvgPool2DOp.h.

{ return _attributes.data_format; }

References mir::AvgPool2DOpAttributes::data_format.

Referenced by mir2loco::Transformer::visit().

getIncludePad()

bool mir::ops::AvgPool2DOp::getIncludePad ( ) const

inline

Definition at line 53 of file AvgPool2DOp.h.

{ return _attributes.include_pad; }

References mir::AvgPool2DOpAttributes::include_pad.

Referenced by mir_interpreter::AvgPool2DImpl< uint8_t >::run(), mir_interpreter::AvgPool2DImpl< T >::run(), mir2loco::Transformer::visit(), nnc::AclCppOpGenerator::visit(), nnc::Serializer::visit(), and mir::DotNodeBuilder::visit().

getPaddingAfter()

const std::vector< std::int32_t > & mir::ops::AvgPool2DOp::getPaddingAfter ( ) const

inline

Definition at line 51 of file AvgPool2DOp.h.

{ return _attributes.padding_after; }

References mir::AvgPool2DOpAttributes::padding_after.

Referenced by mir_interpreter::AvgPool2DImpl< uint8_t >::run(), mir_interpreter::AvgPool2DImpl< T >::run(), mir2loco::Transformer::visit(), and mir::DotNodeBuilder::visit().

getPaddingBefore()

const std::vector< std::int32_t > & mir::ops::AvgPool2DOp::getPaddingBefore ( ) const

inline

Definition at line 49 of file AvgPool2DOp.h.

{ return _attributes.padding_before; }

References mir::AvgPool2DOpAttributes::padding_before.

Referenced by mir_interpreter::AvgPool2DImpl< uint8_t >::run(), mir_interpreter::AvgPool2DImpl< T >::run(), mir2loco::Transformer::visit(), and mir::DotNodeBuilder::visit().

getStrides()

const std::vector< std::int32_t > & mir::ops::AvgPool2DOp::getStrides ( ) const

inline

Definition at line 47 of file AvgPool2DOp.h.

{ return _attributes.strides; }

References mir::AvgPool2DOpAttributes::strides.

Referenced by mir_interpreter::AvgPool2DImpl< uint8_t >::run(), mir_interpreter::AvgPool2DImpl< T >::run(), mir2loco::Transformer::visit(), nnc::Serializer::visit(), and mir::DotNodeBuilder::visit().

getWindowSize()

const std::vector< std::int32_t > & mir::ops::AvgPool2DOp::getWindowSize ( ) const

inline

Definition at line 45 of file AvgPool2DOp.h.

{ return _attributes.window; }

References mir::AvgPool2DOpAttributes::window.

Referenced by mir_interpreter::AvgPool2DImpl< uint8_t >::run(), mir_interpreter::AvgPool2DImpl< T >::run(), mir2loco::Transformer::visit(), nnc::Serializer::visit(), and mir::DotNodeBuilder::visit().

---

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

• compiler/mir/include/mir/ops/AvgPool2DOp.h
• compiler/mir/src/ops/AvgPool2DOp.cpp