mxnet
|
Engine that schedules all the operations according to dependency. More...
#include <dmlc/base.h>
#include <algorithm>
#include <memory>
#include <functional>
#include <vector>
#include "./base.h"
Go to the source code of this file.
Classes | |
class | mxnet::engine::CallbackOnComplete |
OnComplete Callback to the engine, called by AsyncFn when action completes. More... | |
class | mxnet::Engine |
Dependency engine that schedules operations. More... | |
Namespaces | |
mxnet | |
namespace of mxnet | |
mxnet::engine | |
namespace of engine internal types. | |
Typedefs | |
typedef Var * | mxnet::engine::VarHandle |
Variable pointer type, usually hold by user used to specify dependencies. More... | |
typedef Opr * | mxnet::engine::OprHandle |
Operator pointer type, usually hold by user. More... | |
Enumerations | |
enum | mxnet::FnProperty { mxnet::FnProperty::kNormal, mxnet::FnProperty::kCopyFromGPU, mxnet::FnProperty::kCopyToGPU, mxnet::FnProperty::kCPUPrioritized, mxnet::FnProperty::kAsync } |
Function property, used to hint what action is pushed to engine. More... | |
Engine that schedules all the operations according to dependency.