smaug::RepeatOp< Backend > Class Template Reference

Replicates a Tensor's data among all dimensions. More...

#include <repeat_op.h>

Detailed Description

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

Replicates a Tensor's data among all dimensions.

Implements a repeat operator, which replicates the contents of a Tensor along each dimension a configurable number of times. This is set by the setMultiples function.

Template Parameters

Backend

The Backend specialization of this Operator.

Definition at line 44 of file backend.h.

Public Member Functions
	RepeatOp (const std::string &name, Workspace *workspace)
	RepeatOp (const std::string &name, Workspace *workspace, const std::vector< int > _multiples)
void	setMultiples (const std::vector< int > &_multiples)
	Set the number of copies of the Tensor along each dimension.
void	setMultiples (const std::initializer_list< int > &_multiples)
	Set the number of copies of the Tensor along each dimension.
bool	validate () override
void	createAllTensors () override
void	run () override

Protected Attributes
std::vector< int >	multiples

---

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

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