struct dnnl::gru_forward

Overview

GRU forward propagation primitive. More…

#include <dnnl.hpp>

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

    struct desc;
    struct primitive_desc;

    // methods

    gru_forward();
    gru_forward(const primitive_desc& pd);
    gru_forward(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

GRU forward propagation primitive.

Methods

gru_forward()

Default constructor. Produces an empty object.

gru_forward(const primitive_desc& pd)

Constructs a GRU forward propagation primitive.

Parameters:

pd

Primitive descriptor for a GRU forward propagation primitive.

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

Constructs a GRU forward propagation primitive from a cache blob.

Parameters:

pd

Primitive descriptor for a GRU forward propagation primitive.

cache_blob

Cache blob.