struct dnnl::concat::primitive_desc

Overview

Primitive descriptor for a concat primitive. More…

#include <dnnl.hpp>

struct primitive_desc: public dnnl::primitive_desc_base
{
    // methods

    primitive_desc();

    primitive_desc(
        const memory::desc& dst,
        int concat_dimension,
        const std::vector<memory::desc>& srcs,
        const engine& aengine,
        const primitive_attr& attr = primitive_attr()
        );

    primitive_desc(
        int concat_dimension,
        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 concat primitive.

Methods

primitive_desc()

Default constructor. Produces an empty object.

primitive_desc(
    const memory::desc& dst,
    int concat_dimension,
    const std::vector<memory::desc>& srcs,
    const engine& aengine,
    const primitive_attr& attr = primitive_attr()
    )

Constructs a primitive descriptor for an out-of-place concatenation primitive.

Parameters:

dst

Destination memory descriptor.

concat_dimension

Source tensors will be concatenated over dimension with this index. Note that order of dimensions does not depend on memory format.

srcs

Vector of source memory descriptors.

aengine

Engine to perform the operation on.

attr

Primitive attributes to use (optional).

primitive_desc(
    int concat_dimension,
    const std::vector<memory::desc>& srcs,
    const engine& aengine,
    const primitive_attr& attr = primitive_attr()
    )

Constructs a primitive descriptor for an out-of-place concatenation primitive.

This version derives the destination memory descriptor automatically.

Parameters:

concat_dimension

Source tensors will be concatenated over dimension with this index. Note that order of dimensions does not depend on memory format.

srcs

Vector of source memory descriptors.

aengine

Engine to perform the operation on.

attr

Primitive attributes to use (optional).

primitive_desc(dnnl_primitive_desc_t pd)

Constructs a primitive descriptor for concat primitive from a C API primitive descriptor which must have a matching kind.

Parameters:

pd

C API primitive descriptor for concat 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.