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