.rst .pdf

Contents

struct dnnl::shuffle_backward::desc

Overview

Descriptor for a shuffle primitive backward propagation primitive. More…

#include <dnnl.hpp>

struct desc
{
    // fields

    dnnl_shuffle_desc_t data;

    // construction

    desc(const memory::desc& diff_data_desc, int axis, int group_size);
};

Detailed Documentation

Descriptor for a shuffle primitive backward propagation primitive.

Construction

desc(const memory::desc& diff_data_desc, int axis, int group_size)

Constructs a descriptor for a shuffle backward propagation primitive.

Parameters:

diff_data_desc	Diff source and diff destination memory descriptor.
axis	The axis along which the data is shuffled.
group_size	Shuffle group size.

struct dnnl::shuffle_backward struct dnnl::shuffle_backward::primitive_desc