SMAUG
Simulating Machine Learning Applications on gem5-Aladdin
|
Flattens each batch of a Tensor. More...
#include <reorder_op.h>
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.
Public Types | |
typedef ReorderOp< Backend > | Parent |
Public Member Functions | |
FlattenOp (const std::string &name, Workspace *workspace) | |