struct dnnl::sum::primitive_desc¶
Overview¶
Primitive descriptor for a sum primitive. More…
#include <dnnl.hpp> struct primitive_desc: public dnnl::primitive_desc_base { // methods primitive_desc(); primitive_desc( const memory::desc& dst, const std::vector<float>& scales, const std::vector<memory::desc>& srcs, const engine& aengine, const primitive_attr& attr = primitive_attr() ); primitive_desc( const std::vector<float>& scales, const std::vector<memory::desc>& srcs, const engine& aengine, const primitive_attr& attr = primitive_attr() ); primitive_desc(dnnl_primitive_desc_t pd); memory::desc src_desc(int idx = 0) const; memory::desc dst_desc() const; };
Inherited Members¶
public: // 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_desc_base(); engine get_engine() const; const char* impl_info_str() const; memory::dim query_s64(query what) const; memory::desc query_md(query what, int idx = 0) const; memory::desc src_desc(int idx) const; memory::desc dst_desc(int idx) const; memory::desc weights_desc(int idx) const; memory::desc diff_src_desc(int idx) const; memory::desc diff_dst_desc(int idx) const; memory::desc diff_weights_desc(int idx) const; memory::desc src_desc() const; memory::desc dst_desc() const; memory::desc weights_desc() const; memory::desc diff_src_desc() const; memory::desc diff_dst_desc() const; memory::desc diff_weights_desc() const; memory::desc workspace_desc() const; memory::desc scratchpad_desc() const; engine scratchpad_engine() const; primitive_attr get_primitive_attr() const; dnnl::primitive::kind get_kind() const; std::vector<uint8_t> get_cache_blob_id() const;
Detailed Documentation¶
Primitive descriptor for a sum primitive.
Methods¶
primitive_desc()
Default constructor. Produces an empty object.
primitive_desc( const memory::desc& dst, const std::vector<float>& scales, const std::vector<memory::desc>& srcs, const engine& aengine, const primitive_attr& attr = primitive_attr() )
Constructs a primitive descriptor for a sum primitive.
Parameters:
dst |
Destination memory descriptor. |
scales |
Vector of scales to multiply data in each source memory by. |
srcs |
Vector of source memory descriptors. |
aengine |
Engine to perform the operation on. |
attr |
Primitive attributes to use (optional). |
primitive_desc( const std::vector<float>& scales, const std::vector<memory::desc>& srcs, const engine& aengine, const primitive_attr& attr = primitive_attr() )
Constructs a primitive descriptor for a sum primitive.
This version derives the destination memory descriptor automatically.
Parameters:
scales |
Vector of scales by which to multiply data in each source memory object. |
srcs |
Vector of source memory descriptors. |
aengine |
Engine on which to perform the operation. |
attr |
Primitive attributes to use (optional). |
primitive_desc(dnnl_primitive_desc_t pd)
Constructs a primitive descriptor for sum primitive from a C API primitive descriptor which must have a matching kind.
Parameters:
pd |
C API primitive descriptor for reorder primitive. |
memory::desc src_desc(int idx = 0) const
Returns a source memory descriptor.
Parameters:
idx |
Source index. |
Returns:
Source memory descriptor.
A zero memory descriptor if the primitive does not have a source parameter with index idx
.
memory::desc dst_desc() const
Returns a destination memory descriptor.
Returns:
Destination memory descriptor.
A zero memory descriptor if the primitive does not have a destination parameter.