oneAPI Deep Neural Network Library (oneDNN)
Performance library for Deep Learning
1.96.0
dnnl::shuffle_backward::desc Struct Reference

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

#include <dnnl.hpp>

Collaboration diagram for dnnl::shuffle_backward::desc:

Public Member Functions

 desc (const memory::desc &diff_data_desc, int axis, int group_size)
 Constructs a descriptor for a shuffle backward propagation primitive. More...
 

Detailed Description

Descriptor for a shuffle primitive backward propagation primitive.

Constructor & Destructor Documentation

◆ desc()

dnnl::shuffle_backward::desc::desc ( const memory::desc diff_data_desc,
int  axis,
int  group_size 
)
inline

Constructs a descriptor for a shuffle backward propagation primitive.

Parameters
diff_data_descDiff source and diff destination memory descriptor.
axisThe axis along which the data is shuffled.
group_sizeShuffle group size.

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