smaug::ReorderOp< Backend > Class Template Reference

Implements a Tensor reordering operation to convert between different DataLayouts. More...

#include <reorder_op.h>

Detailed Description

template<typename Backend>
class smaug::ReorderOp< Backend >

Implements a Tensor reordering operation to convert between different DataLayouts.

Template Parameters

Backend

The Backend specialization of this Operator.

Definition at line 40 of file backend.h.

Public Member Functions
	ReorderOp (const std::string &name, DataLayout _targetLayout, Workspace *workspace)
	ReorderOp (const std::string &name, Workspace *workspace)
DataLayout	getTargetDataLayout () const
void	setTargetLayout (DataLayout layout)
void	run () override
bool	validate () override
TensorShape	inferOutputShape () const
void	createOutputTensors ()
void	createAllTensors () override

Protected Types
enum	{ Inputs, kNumInputs }
enum	{ Outputs, kNumOutputs }

Protected Attributes
DataLayout	targetLayout

---

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

• smaug/core/backend.h
• smaug/operators/reorder_op.h