.rst .pdf

Contents

struct dnnl::reorder

Overview

Reorder primitive. More…

#include <dnnl.hpp>

struct reorder: public dnnl::primitive
{
    // structs

    struct primitive_desc;

    // methods

    reorder();
    reorder(const primitive_desc& pd);
    reorder(const primitive_desc& pd, const std::vector<uint8_t>& cache_blob);

    reorder(
        const memory& src,
        const memory& dst,
        const primitive_attr& attr = primitive_attr()
        );

    void execute(const stream& astream, memory& src, memory& dst) const;
};

Inherited Members

public:
    // enums

    enum kind;

    // methods

    handle();
    handle(const handle<T, traits>&);
    handle<T, traits>& operator = (const handle<T, traits>&);
    handle(handle<T, traits>&&);
    handle<T, traits>& operator = (handle<T, traits>&&);
    handle(T t, bool weak = false);
    void reset(T t, bool weak = false);
    T get(bool allow_empty = false) const;
    operator T () const;
    operator bool () const;
    bool operator == (const handle<T, traits>& other) const;
    bool operator != (const handle& other) const;
    primitive();
    primitive(const_dnnl_primitive_desc_t c_pd);

    primitive(
        const_dnnl_primitive_desc_t c_pd,
        const std::vector<uint8_t>& cache_blob
        );

    primitive(const primitive_desc& pd);
    primitive(const primitive_desc& pd, const std::vector<uint8_t>& cache_blob);
    const_dnnl_primitive_desc_t get_primitive_desc() const;
    kind get_kind() const;
    std::vector<uint8_t> get_cache_blob() const;
    void execute(const stream& astream, const std::unordered_map<int, memory>& args) const;

Detailed Documentation

Reorder primitive.

Methods

reorder()

Default constructor. Produces an empty object.

reorder(const primitive_desc& pd)

Constructs a reorder primitive.

Parameters:

pd	Primitive descriptor for reorder primitive.

reorder(const primitive_desc& pd, const std::vector<uint8_t>& cache_blob)

Constructs a reorder primitive from a cache blob.

Parameters:

pd	Primitive descriptor for reorder primitive.
cache_blob	Cache blob.

reorder(
    const memory& src,
    const memory& dst,
    const primitive_attr& attr = primitive_attr()
    )

Constructs a reorder primitive that would reorder data between memory objects having the same memory descriptors as memory objects src and dst.

Parameters:

src	Source memory object.
dst	Destination memory object.
attr	Primitive attributes to use (optional).

void execute(const stream& astream, memory& src, memory& dst) const

Executes the reorder primitive.

Parameters:

astream	Stream object. The stream must belong to the same engine as the primitive.
src	Source memory object.
dst	Destination memory object.

Reorder struct dnnl::reorder::primitive_desc