ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Public Member Functions | Protected Member Functions
mir::ops::ReduceOp Class Reference

#include <ReduceOp.h>

Collaboration diagram for mir::ops::ReduceOp:
Collaboration graph
[legend]

Public Member Functions

const std::vector< int > & getReductionDims () const
 
bool getKeepDims () 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)
 
virtual Operation * copyWithInputs (const std::vector< Output * > &inputs)=0
 

Protected Member Functions

 ReduceOp (Type type, Output *arg, const std::vector< int > &reduction_dims, bool keep_dims)
 
- 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)
 

Additional Inherited Members

- Public Types inherited from mir::Operation
enum class  Type { HANDLE_OP }
 

Detailed Description

Definition at line 28 of file ReduceOp.h.

Constructor & Destructor Documentation

◆ ReduceOp()

mir::ops::ReduceOp::ReduceOp ( Type  type,
Output *  arg,
const std::vector< int > &  reduction_dims,
bool  keep_dims 
)
inlineprotected

Definition at line 31 of file ReduceOp.h.

32 : Operation(type, {arg}), _reduction_dims(reduction_dims), _keep_dims(keep_dims)
33 {
34 inferOutputTypes();
35 }
infer_testcases.arg
arg
Definition infer_testcases.py:46

Member Function Documentation

◆ getKeepDims()

bool mir::ops::ReduceOp::getKeepDims ( ) const
inline

Definition at line 40 of file ReduceOp.h.

40{ return _keep_dims; };

Referenced by mir::ops::ReduceMeanOp::copyWithInputs(), mir_interpreter::ReduceMeanImpl< T >::run(), nnc::Serializer::visit(), and mir::DotNodeBuilder::visit().

◆ getReductionDims()

const std::vector< int > & mir::ops::ReduceOp::getReductionDims ( ) const
inline

Definition at line 38 of file ReduceOp.h.

38{ return _reduction_dims; };

Referenced by mir::ops::ReduceMeanOp::copyWithInputs(), mir_interpreter::ReduceMeanImpl< T >::run(), nnc::Serializer::visit(), and mir::DotNodeBuilder::visit().

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