Automatic Layout Manager. More...
#include <mxnet/base.h>#include <nnvm/graph.h>#include <nnvm/node.h>#include <functional>#include <string>#include <unordered_map>#include <vector>
Include dependency graph for alm.h:
Go to the source code of this file.
Classes | |
| struct | mxnet::alm::ALMParams |
| A singleton flag, set and read by MXSetOptimizeLayout and MXGetOptimizeLayout. More... | |
Namespaces | |
| mxnet | |
| namespace of mxnet | |
| mxnet::alm | |
Typedefs | |
| using | mxnet::alm::Transpose = std::vector< size_t > |
| Transpose, represented by permutation of axes. More... | |
| using | mxnet::alm::FChangeLayout = std::function< bool(nnvm::NodeAttrs *, mshadow::LayoutFlag target_layout, std::vector< Transpose > *in_axes, std::vector< Transpose > *out_axes)> |
| May change operator's layout. Used in LayoutOptimization. More... | |
Functions | |
| nnvm::Graph | mxnet::alm::OptimizeLayout (nnvm::Graph &&g) |
| bool | mxnet::alm::IsIdentity (const Transpose &t) |
| Transpose | mxnet::alm::Reverse (const Transpose &axes) |
| Transpose | mxnet::alm::Compose (const Transpose &lhs, const Transpose &rhs) |
| mshadow::LayoutFlag | mxnet::alm::ApplyTranspose (mshadow::LayoutFlag layout, const Transpose &axes) |
| std::string | mxnet::alm::ApplyTranspose (const std::string &layout, const Transpose &axes) |
| Transpose | mxnet::alm::FromTShape (const mxnet::TShape &s) |
| Transpose | mxnet::alm::FactorCommonTranspose (std::vector< Transpose > *axes) |
| Factors out and returns a common transpose, or default-constructed Transpose if all axes (in/out parameter) are empty. More... | |
Detailed Description
Automatic Layout Manager.
