.. SPDX-FileCopyrightText: 2019-2020 Intel Corporation .. .. SPDX-License-Identifier: CC-BY-4.0 .. default-domain:: cpp .. include:: ../replacements.inc.rst ####### Reorder ####### A primitive to copy data between two memory objects. This primitive is typically used to change the way that the data is laid out in memory. The reorder primitive copies data between different memory formats but does not change the tensor from mathematical perspective. Variable names follow the standard :ref:`conventions-label`. .. math:: \dst(\overline{x}) = \src(\overline{x}) for :math:`\overline{x} = (x_0, \ldots, x_n)`. As described in :ref:`introduction-label` in order to achieve the best performance some primitives (such as convolution) require special memory format which is typically referred to as an *optimized* memory format. The *optimized* memory format may match or may not match memory format that data is currently kept in. In this case a user can use reorder primitive to copy (reorder) the data between the memory formats. Using the attributes and post-ops users can also use reorder primitive to quantize the data (and if necessary change the memory format simultaneously). ******************* Execution Arguments ******************* When executed, the inputs and outputs should be mapped to an execution argument index as specified by the following table. ====================== ======================== Primitive input/output Execution argument index ====================== ======================== :math:`\src` |DNNL_ARG_FROM| :math:`\dst` |DNNL_ARG_TO| ====================== ======================== ***************** Operation Details ***************** 1. The reorder primitive requires the source and destination tensors to have the same shape. Implicit broadcasting is not supported. 2. While in most of the cases the reorder should be able to handle arbitrary source and destination memory formats and data types, it might happen than some combinations are not implemented. For instance: - Reorder implementations between weights in non-plain memory formats might be limited (but if encountered in real practice should be treated as a bug and reported to oneDNN team); - Weights in one Winograd format cannot be reordered to the weights of the other Winograd format; - Quantized weights for convolution with #dnnl_s8 source data type cannot be dequantized back to the #dnnl_f32 data type; 3. To alleviate the problem a user may rely on fact that the reorder from original plain memory format and user's data type to the *optimized* format with chosen data type should be always implemented. ****************** Data Types Support ****************** The reorder primitive supports arbitrary data types for the source and destination. When converting the data from one data type to a smaller one saturation is used. For instance: :: reorder(src={1024, data_type=f32}, dst={, data_type=s8}) // dst == {127} reorder(src={-124, data_type=f32}, dst={, data_type=u8}) // dst == {0} ******************* Data Representation ******************* The reorder primitive works with arbitrary data tensors. There is no special meaning associated with any logical dimensions. *********************** Post-ops and Attributes *********************** The reorder primitive should support the following attributes and post-ops: +-----------+---------------------------------------------------------------------+-------------------------------------------------------------------------------+------------------------+ | Type | Operation | Description | Restrictions | +===========+=====================================================================+===============================================================================+========================+ | Attribute | :any:`Scales ` | Sets scale(s) for the corresponding tensor(s) | Int8 computations only | +-----------+---------------------------------------------------------------------+-------------------------------------------------------------------------------+------------------------+ | Attribute | :any:`Zero points ` | Sets zero point(s) for the corresponding tensors | Int8 computations only | +-----------+---------------------------------------------------------------------+-------------------------------------------------------------------------------+------------------------+ | post-op | :any:`Sum ` | Adds the operation result to the destination tensor instead of overwriting it | | +-----------+---------------------------------------------------------------------+-------------------------------------------------------------------------------+------------------------+ For instance, the following pseudo-code :: reorder( src = {dims={N, C, H, W}, data_type=dt_src, memory_format=fmt_src}, dst = {dims={N, C, H, W}, data_type=dt_dst, memory_format=fmt_dst}, attr ={ output_scale=alpha, post-ops = { sum={scale=beta} }, }) would lead to the following operation: .. math:: \dst(\overline{x}) = \alpha \cdot \src(\overline{x}) + \beta \cdot \dst(\overline{x}) .. note:: The intermediate operations are being done using single precision floating point data type. *** API *** .. doxygenstruct:: dnnl::reorder :project: oneDNN :members: .. vim: ts=3 sw=3 et spell spelllang=en