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;

    // methods

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

Detailed Documentation

Descriptor for a shuffle primitive backward propagation primitive.

Methods

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.