mxnet
|
#include <mxnet/op_attr_types.h>
#include <mxnet/graph_attr_types.h>
#include <mxnet/c_api.h>
#include <nnvm/symbolic.h>
#include <nnvm/op.h>
#include <nnvm/graph.h>
#include <vector>
#include <atomic>
#include <utility>
#include <string>
#include <unordered_map>
#include "./ndarray.h"
Go to the source code of this file.
Classes | |
class | mxnet::Imperative |
runtime functions for NDArray More... | |
class | mxnet::Imperative::AGInfo |
class | mxnet::Imperative::DCInfo |
DCInfo datastructure to enable deferred computation. More... | |
Namespaces | |
mxnet | |
namespace of mxnet | |
Typedefs | |
using | mxnet::OptConstraint_int_t = std::underlying_type_t< OptConstraint > |
typedef NumpyShape | mxnet::NumpyDefaultDtype |
Enumerations | |
enum | mxnet::OptConstraint : unsigned int { mxnet::OptConstraint::None = 0, mxnet::OptConstraint::DisableAMP = 1 << 0 } |
enum | mxnet::NumpyShape { mxnet::Off, mxnet::ThreadLocalOn, mxnet::GlobalOn } |
there are three numpy shape flags based on priority. GlobalOn turn on numpy shape flag globally, it includes thread local. The flag can be seen in any thread. ThreadLocalOn only turn on thread local numpy shape flag, it cannot be seen in other threads. Off turn off numpy shape flag globally. More... | |
Variables | |
constexpr char | mxnet::OPT_CONSTRAINT_ATTR [] = "__opt_constraint__" |