mxnet
|
Engine that schedules all the operations according to dependency. More...
#include <algorithm>
#include <memory>
#include <functional>
#include <utility>
#include <vector>
#include "./base.h"
Go to the source code of this file.
Classes | |
class | mxnet::engine::CUDAEvent |
class | mxnet::engine::CUDAEventPool |
struct | mxnet::engine::EventInfo |
full event info for the sync object. More... | |
struct | mxnet::engine::SyncObject |
struct containing cuda events and variables needed for the dependencies. More... | |
struct | mxnet::engine::Var |
base class of engine variables. More... | |
class | mxnet::engine::CallbackOnStart |
OnStart callback to the engine, called by AsyncFn before the action. More... | |
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, mxnet::FnProperty::kDeleteVar, mxnet::FnProperty::kGPUPrioritized, mxnet::FnProperty::kNoSkip } |
Function property, used to hint what action is pushed to engine. More... | |
Engine that schedules all the operations according to dependency.