.rst .pdf

Contents

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;

    // construction

    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.

Construction

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.

struct dnnl::inner_product_backward_weights struct dnnl::inner_product_backward_weights::primitive_desc