.rst .pdf

Contents

struct dnnl::shuffle_forward::desc

Overview

Descriptor for a shuffle forward propagation primitive. More…

#include <dnnl.hpp>

struct desc
{
    // fields

    dnnl_shuffle_desc_t data;

    // construction

    desc(
        prop_kind aprop_kind,
        const memory::desc& data_desc,
        int axis,
        int group_size
        );
};

Detailed Documentation

Descriptor for a shuffle forward propagation primitive.

Construction

desc(
    prop_kind aprop_kind,
    const memory::desc& data_desc,
    int axis,
    int group_size
    )

Constructs a descriptor for a shuffle forward propagation primitive.

Parameters:

aprop_kind	Propagation kind. Possible values are dnnl::prop_kind::forward_training, and dnnl::prop_kind::forward_inference.
data_desc	Source and destination memory descriptor.
axis	The axis along which the data is shuffled.
group_size	Shuffle group size.

struct dnnl::shuffle_forward struct dnnl::shuffle_forward::primitive_desc