mir::ops::ReshapeOp Class Reference

#include <ReshapeOp.h>

Collaboration diagram for mir::ops::ReshapeOp:

Public Member Functions
	ReshapeOp (Output *arg, const Shape &shape)
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 ReshapeOp.h.

Constructor & Destructor Documentation

ReshapeOp()

mir::ops::ReshapeOp::ReshapeOp ( Output *  arg, const Shape &  shape  )

inline

Definition at line 30 of file ReshapeOp.h.

                                             : Operation(Type::reshape, {arg})
  {
    const Shape &input_shape = getInputShape(0);
    auto output_shape = shape;
 
    auto in_elements_num = input_shape.numElements();
    int32_t out_elements_num = 1;
    // Can't use num_elements due to -1 in input shape and Shape using unsigned ints for dimensions.
    for (int32_t d = 0; d < output_shape.rank(); ++d)
    {
      auto dim = output_shape.dim(d);
      if (dim != Shape::autoDim)
        out_elements_num *= dim;
    }
 
    for (int32_t d = 0; d < output_shape.rank(); ++d)
    {
      auto &dim = output_shape.dim(d);
      if (dim == Shape::autoDim)
        dim = static_cast<int32_t>(in_elements_num / out_elements_num);
    }
 
    setOutputType(0, {arg->getElementType(), output_shape});
  }

Member Function Documentation

copyWithInputs()

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

inlineoverridevirtual

Implements mir::Operation.

Definition at line 55 of file ReshapeOp.h.

  {
    return new ReshapeOp(inputs[0], getOutputShape(0));
  }

References mir::Operation::getOutputShape().

---

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

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