Detailed Description

Concatenates N Tensors along a specified axis.

This has a software-based implementation.

Template Parameters

Backend The Backend that sets Alignment.

Definition at line 41 of file backend.h.

Public Member Functions
	ConcatOp (const std::string &name, Workspace *workspace)

	ConcatOp (const std::string &name, Workspace *workspace, int num, int axis=0)
	Create a ConcatOp. More...

void	setNumInputs (int num)
	Set the number of Tensors to concatenate.

void	setConcatAxis (int axis)
	Set the axis along which to concatenate.

TensorShape	inferOutputShape () const

void	createOutputTensor ()

void	createAllTensors () override

void	run () override

int	getConcatAxis () const

Constructor & Destructor Documentation

template<typename Backend >

smaug::ConcatOp< Backend >::ConcatOp	(	const std::string &	name,
		Workspace *	workspace,
		int	num,
		int	axis = `0`
	)

inline

Create a ConcatOp.

Parameters

name	Operator name
workspace	Workspace to manage this Operator.
num	Number of tensors to concatenate.
axis	Axis/dimension along which to concatenate.

Definition at line 33 of file concat_op.h.

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