A primitive to shuffle tensor data along an axis.
More...
A primitive to shuffle tensor data along an axis.
- See also
- Shuffle in developer guide
◆ dnnl_shuffle_forward_desc_init()
Initializes a descriptor for shuffle forward propagation primitive.
- Parameters
-
| shuffle_desc | Output descriptor for a shuffle primitive. |
| prop_kind | Propagation kind. Possible values are dnnl_forward_training and dnnl_forward_inference. |
| data_desc | Source and destination memory descriptor. |
| axis | The axis along which the data is shuffled. |
| group_size | Shuffle group size. |
- Returns
- dnnl_success on success and a status describing the error otherwise.
◆ dnnl_shuffle_backward_desc_init()
Initializes a descriptor for shuffle backward propagation primitive.
- Parameters
-
| shuffle_desc | Output descriptor for a shuffle primitive. |
| diff_data_desc | Diff source and diff destination memory descriptor. |
| axis | The axis along which the data is shuffled. |
| group_size | Shuffle group size. |
- Returns
- dnnl_success on success and a status describing the error otherwise.