struct dnnl::concat

Overview

Tensor concatenation (concat) primitive. More…

#include <dnnl.hpp>

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

    struct primitive_desc;

    // methods

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

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

Tensor concatenation (concat) primitive.

Methods

concat()

Default constructor. Produces an empty object.

concat(const primitive_desc& pd)

Constructs a concatenation primitive.

Parameters:

pd

Primitive descriptor for concatenation primitive.

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

Constructs a concatenation primitive from a cache blob.

Parameters:

pd

Primitive descriptor for concatenation primitive.

cache_blob

Cache blob.