description of softmax operation. More...

#include <SoftmaxOp.h>

Collaboration diagram for mir::ops::SoftmaxOp:

Public Member Functions
	SoftmaxOp (Output *arg, int32_t axis)

Operation *	copyWithInputs (const std::vector< Output * > &inputs) override

int32_t	getAxis () 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

description of softmax operation.

Definition at line 30 of file SoftmaxOp.h.

Constructor & Destructor Documentation

◆ SoftmaxOp()

mir::ops::SoftmaxOp::SoftmaxOp	(	Output *	arg,
		int32_t	axis
	)

inline

Definition at line 33 of file SoftmaxOp.h.

                                       : Operation(Type::softmax, {arg}), _axis(axis)
  {
    setOutputType(0, {arg->getElementType(), arg->getShape()});
  }

Member Function Documentation

◆ copyWithInputs()

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

inlineoverridevirtual

Implements mir::Operation.

Definition at line 38 of file SoftmaxOp.h.

  {
    return new SoftmaxOp(inputs[0], _axis);
  }

◆ getAxis()

int32_t mir::ops::SoftmaxOp::getAxis ( ) const

inline

Definition at line 43 of file SoftmaxOp.h.

  {
    if (_axis < 0)
    {
      // Negative axis is used to index starting from the last element of the shape
      // -1 means last element, -2 means second from end, like in python
      int32_t res = _axis + getInputShape(0).rank();
      assert(res >= 0);
      return res;
    }
    return _axis;
  }

References mir::Operation::getInputShape(), and mir::Shape::rank().

Referenced by TEST(), mir_interpreter::MIRInterpreter::visit(), mir2loco::Transformer::visit(), nnc::AclCppOpGenerator::visit(), nnc::Serializer::visit(), and mir::DotNodeBuilder::visit().

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

compiler/mir/include/mir/ops/SoftmaxOp.h