#include <Deconv2DOp.h>

Collaboration diagram for mir::ops::DeConv2DOp:

Public Member Functions
	DeConv2DOp (Output input, Output kernel, const Deconv2DOpAttributes &attributes)

	DeConv2DOp (Output input, Output kernel, const Deconv2DOpAttributes &attributes, const Shape &output_shape)

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

const std::vector< std::int32_t > &	getStrides () const

PaddingType	getPaddingType () const

const std::vector< std::int32_t > &	getPaddingBefore () const

const std::vector< std::int32_t > &	getPaddingAfter () const

DataFormat	getDataFormat () const

const Deconv2DOpAttributes &	getAttributes () 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

Definition at line 32 of file Deconv2DOp.h.

Constructor & Destructor Documentation

◆ DeConv2DOp() [1/2]

mir::ops::DeConv2DOp::DeConv2DOp	(	Output *	input,
		Output *	kernel,
		const Deconv2DOpAttributes &	attributes
	)

inline

Definition at line 35 of file Deconv2DOp.h.

    : Operation(Type::deConv2D, {input, kernel}), _attributes(attributes)
  {
    inferOutputTypes();
  }

◆ DeConv2DOp() [2/2]

mir::ops::DeConv2DOp::DeConv2DOp	(	Output *	input,
		Output *	kernel,
		const Deconv2DOpAttributes &	attributes,
		const Shape &	output_shape
	)

inline

Definition at line 41 of file Deconv2DOp.h.

    : Operation(Type::deConv2D, {input, kernel}), _attributes(attributes)
  {
    assert(input->getElementType() == kernel->getElementType());
    setOutputType(0, {input->getElementType(), output_shape});
    inferPaddings();
  }

Member Function Documentation

◆ copyWithInputs()

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

inlineoverridevirtual

Implements mir::Operation.

Definition at line 50 of file Deconv2DOp.h.

  {
    if (getPaddingType() == PaddingType::Explicit)
      return new DeConv2DOp(inputs[0], inputs[1], _attributes);
    else
      return new DeConv2DOp(inputs[0], inputs[1], _attributes, getOutputShape(0));
  }