SMAUG
Simulating Machine Learning Applications on gem5-Aladdin
Public Types | Public Member Functions | List of all members
smaug::FlattenOp< Backend > Class Template Reference

Flattens each batch of a Tensor. More...

#include <reorder_op.h>

Detailed Description

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

Flattens each batch of a Tensor.

Implements a flattening operation that squashes any Tensor DataLayout into NC; that is, each batch in a Tensor is flattened into a single dimension.

Definition at line 45 of file backend.h.

Public Types

typedef ReorderOp< Backend > Parent
 

Public Member Functions

 FlattenOp (const std::string &name, Workspace *workspace)
 

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