struct dnnl::inner_product_backward_weights::desc

Overview

Descriptor for an inner product weights gradient primitive. More…

#include <dnnl.hpp>

struct desc
{
    // fields

    dnnl_inner_product_desc_t data;

    // methods

    desc(
        const memory::desc& src_desc,
        const memory::desc& diff_weights_desc,
        const memory::desc& diff_bias_desc,
        const memory::desc& diff_dst_desc
        );

    desc(
        const memory::desc& src_desc,
        const memory::desc& diff_weights_desc,
        const memory::desc& diff_dst_desc
        );
};

Detailed Documentation

Descriptor for an inner product weights gradient primitive.

Methods

desc(
    const memory::desc& src_desc,
    const memory::desc& diff_weights_desc,
    const memory::desc& diff_bias_desc,
    const memory::desc& diff_dst_desc
    )

Constructs a descriptor for an inner product descriptor weights update primitive with bias.

Note

All the memory descriptors may be initialized with the dnnl::memory::format_tag::any value of format_tag.

Parameters:

src_desc

Memory descriptor for src.

diff_weights_desc

Memory descriptor for diff weights.

diff_bias_desc

Memory descriptor for diff bias.

diff_dst_desc

Memory descriptor for diff dst.

desc(
    const memory::desc& src_desc,
    const memory::desc& diff_weights_desc,
    const memory::desc& diff_dst_desc
    )

Constructs a descriptor for an inner product descriptor weights update primitive without bias.

Note

All the memory descriptors may be initialized with the dnnl::memory::format_tag::any value of format_tag.

Parameters:

src_desc

Memory descriptor for src.

diff_weights_desc

Memory descriptor for diff weights.

diff_dst_desc

Memory descriptor for diff dst.