mxnet
Classes | Namespaces | Typedefs | Enumerations
engine.h File Reference

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"
Include dependency graph for engine.h:
This graph shows which files directly or indirectly include this file:

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

Detailed Description

Engine that schedules all the operations according to dependency.