#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.
|
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...
|
|