Collaboration diagram for mir::ops::BroadcastOp:

Public Member Functions
	BroadcastOp (Output *input, const Shape &target_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 BroadcastOp.h.

Constructor & Destructor Documentation

mir::ops::BroadcastOp::BroadcastOp	(	Output *	input,
		const Shape &	target_shape
	)

inline

Definition at line 30 of file BroadcastOp.h.

                                                        : Operation(Type::broadcast, {input})
  {
    inferOutputTypes(target_shape);
  }

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

inlineoverridevirtual

Definition at line 35 of file BroadcastOp.h.

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

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