An execution engine. More...

#include <mkldnn.hpp>

Inheritance diagram for mkldnn::engine:

Collaboration diagram for mkldnn::engine:

Public Types
enum	kind { kind::any = mkldnn_any_engine, kind::cpu = mkldnn_cpu, kind::gpu = mkldnn_gpu }
	Kinds of engines. More...

Public Member Functions
	engine (kind akind, size_t index)
	Constructs an engine. More...

	engine (const mkldnn_engine_t &aengine)
	Constructs an engine from other engine `aengine`.

	engine (const handle< mkldnn_primitive_desc_t > &pd)
	Constructs an engine from the primitive descriptor `pd` by querying its engine. More...

kind	get_kind () const
	Returns the kind of the engine.

Public Member Functions inherited from mkldnn::handle< mkldnn_engine_t >
	handle (mkldnn_engine_t t, bool weak=false)
	Constructs a C handle wrapper. More...

	handle ()
	Empty constructor. More...

void	reset (mkldnn_engine_t t, bool weak=false)
	Resets the value of a C handle. More...

mkldnn_engine_t	get (bool allow_emtpy=false) const
	Returns the value of the underlying C handle.

Static Public Member Functions
static size_t	get_count (kind akind)
	Returns the number of engines of a certain kind. More...

Detailed Description

An execution engine.

Examples:: cpu_cnn_inference_f32.cpp, cpu_getting_started.cpp, cpu_memory_format_propagation.cpp, cpu_performance_profiling.cpp, and gpu_opencl_interop.cpp.

Member Enumeration Documentation

strong

Kinds of engines.

inline

Constructs an engine.

Parameters

akind	The kind of engine to construct.
index	The index of the engine. Must be less than the value returned by get_count() for this particular kind of engine.

mkldnn::engine::engine ( const handle< mkldnn_primitive_desc_t > & pd )

inline

Constructs an engine from the primitive descriptor pd by querying its engine.

static size_t mkldnn::engine::get_count ( kind akind )

inlinestatic

Returns the number of engines of a certain kind.

Parameters

akind The kind of engines to count.

The documentation for this struct was generated from the following file: