mir::ops::SigmoidOp Class Reference

#include <SigmoidOp.h>

Collaboration diagram for mir::ops::SigmoidOp:

Public Member Functions
	SigmoidOp (Output *arg)
Operation *	copyWithInputs (const std::vector< Output * > &inputs) override
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 27 of file SigmoidOp.h.

Constructor & Destructor Documentation

SigmoidOp()

mir::ops::SigmoidOp::SigmoidOp ( Output *  arg )

inlineexplicit

Definition at line 30 of file SigmoidOp.h.

                                  : Operation(Type::sigmoid, {arg})
  {
    // Infer output shape.
    setOutputType(0, {arg->getElementType(), arg->getShape()});
  }

Member Function Documentation

copyWithInputs()

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

inlineoverridevirtual

Implements mir::Operation.

Definition at line 36 of file SigmoidOp.h.

  {
    return new SigmoidOp(inputs[0]);
  }

---

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

• compiler/mir/include/mir/ops/SigmoidOp.h