.rst .pdf

Contents

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.

struct dnnl::sum Binary