struct dnnl::softmax_backward

Overview

Softmax backward propagation primitive. More…

#include <dnnl.hpp>

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

    struct primitive_desc;

    // construction

    softmax_backward();
    softmax_backward(const primitive_desc& pd);

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

Inherited Members

public:
    // enums

    enum kind;

    // methods

    handle<T, traits>& operator = (const handle<T, traits>&);
    handle<T, traits>& operator = (handle<T, traits>&&);
    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;
    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;
    handle();
    handle();
    handle();
    handle();

Detailed Documentation

Softmax backward propagation primitive.

Construction

softmax_backward()

Default constructor. Produces an empty object.

softmax_backward(const primitive_desc& pd)

Constructs a softmax backward propagation primitive.

Parameters:

pd

Primitive descriptor for a softmax backward propagation primitive.

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

Constructs a softmax backward propagation primitive from a cache blob.

Parameters:

pd

Primitive descriptor for a softmax backward propagation primitive.

cache_blob

Cache blob.