SMAUG
Simulating Machine Learning Applications on gem5-Aladdin
|
This is the complete list of members for smaug::SplitOp< Backend >, including all inherited members.
createAllTensors() override (defined in smaug::SplitOp< Backend >) | smaug::SplitOp< Backend > | inline |
getSplitAxis() const (defined in smaug::SplitOp< Backend >) | smaug::SplitOp< Backend > | inline |
getSplits() const (defined in smaug::SplitOp< Backend >) | smaug::SplitOp< Backend > | inline |
run() override (defined in smaug::SplitOp< Backend >) | smaug::SplitOp< Backend > | inline |
setSplitAxis(int axis) | smaug::SplitOp< Backend > | inline |
setSplits(const std::vector< int > &_splits) | smaug::SplitOp< Backend > | inline |
setSplits(const std::initializer_list< int > &_splits) (defined in smaug::SplitOp< Backend >) | smaug::SplitOp< Backend > | inline |
splitAxis (defined in smaug::SplitOp< Backend >) | smaug::SplitOp< Backend > | protected |
SplitOp(const std::string &name, Workspace *workspace) (defined in smaug::SplitOp< Backend >) | smaug::SplitOp< Backend > | inline |
SplitOp(const std::string &name, Workspace *workspace, const std::vector< int > &_splits, int axis=0) (defined in smaug::SplitOp< Backend >) | smaug::SplitOp< Backend > | inline |
splits (defined in smaug::SplitOp< Backend >) | smaug::SplitOp< Backend > | protected |
validate() override (defined in smaug::SplitOp< Backend >) | smaug::SplitOp< Backend > | inline |