dnnl::shuffle_forward::desc Struct Reference

Descriptor for a shuffle forward propagation primitive. More...

#include <dnnl.hpp>

Collaboration diagram for dnnl::shuffle_forward::desc:

Public Member Functions
	desc (prop_kind prop_kind, const memory::desc &data_desc, int axis, int group_size)
	Constructs a descriptor for a shuffle forward propagation primitive.

Detailed Description

Descriptor for a shuffle forward propagation primitive.

Examples

shuffle.cpp.

Constructor & Destructor Documentation

desc()

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

inline

Constructs a descriptor for a shuffle forward propagation primitive.

Inputs:

• src (dnnl::primitive_desc_base::src_desc(0))

Outputs:

• dst (dnnl::primitive_desc_base::dst_desc(0))

Parameters

prop_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.

---

The documentation for this struct was generated from the following file:

• include/dnnl.hpp