Engine that schedules all the operations according to dependency. More...

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

Classes
struct	mxnet::engine::Var
	base class of engine variables. 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 }
	Function property, used to hint what action is pushed to engine. More...

Detailed Description

Engine that schedules all the operations according to dependency.

Classes