ONE - On-device Neural Engine
Loading...
Searching...
No Matches
mir::ops::CappedReluOp Class Reference

#include <CappedReluOp.h>

Collaboration diagram for mir::ops::CappedReluOp:

Public Member Functions

 CappedReluOp (Output *arg, float cap)
 
OperationcopyWithInputs (const std::vector< Output * > &inputs) override
 
float getCap () 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
 
OutputgetInput (std::size_t index)
 
const OutputgetInput (std::size_t index) const
 
OutputgetOutput (std::size_t index)
 
const OutputgetOutput (std::size_t index) const
 
const ShapegetInputShape (std::size_t index) const
 
const ShapegetOutputShape (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 27 of file CappedReluOp.h.

Constructor & Destructor Documentation

◆ CappedReluOp()

mir::ops::CappedReluOp::CappedReluOp ( Output arg,
float  cap 
)
inline

Definition at line 30 of file CappedReluOp.h.

30 : Operation(Type::cappedReLU, {arg}), _cap(cap)
31 {
32 // Infer output shape.
33 setOutputType(0, {arg->getElementType(), arg->getShape()});
34 }
void setOutputType(std::size_t index, const TensorType &type)
Definition Operation.h:172

Member Function Documentation

◆ copyWithInputs()

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

Implements mir::Operation.

Definition at line 36 of file CappedReluOp.h.

37 {
38 return new CappedReluOp(inputs[0], _cap);
39 }
CappedReluOp(Output *arg, float cap)

◆ getCap()

float mir::ops::CappedReluOp::getCap ( ) const
inline

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