mxnet
|
Functions | |
mkldnn_status_t MKLDNN_API | mkldnn_reorder_primitive_desc_create (mkldnn_primitive_desc_t *reorder_primitive_desc, const_mkldnn_primitive_desc_t input, const_mkldnn_primitive_desc_t output) |
mkldnn_status_t MKLDNN_API | mkldnn_reorder_primitive_desc_create_v2 (mkldnn_primitive_desc_t *reorder_primitive_desc, const_mkldnn_primitive_desc_t input, const_mkldnn_primitive_desc_t output, const_mkldnn_primitive_attr_t attr) |
A primitive to copy data between memory formats.
mkldnn_status_t MKLDNN_API mkldnn_reorder_primitive_desc_create | ( | mkldnn_primitive_desc_t * | reorder_primitive_desc, |
const_mkldnn_primitive_desc_t | input, | ||
const_mkldnn_primitive_desc_t | output | ||
) |
Initializes a reorder_primitive_desc
using descriptors of input
and output
memory primitives.
Order of inputs:
Order of outputs:
mkldnn_status_t MKLDNN_API mkldnn_reorder_primitive_desc_create_v2 | ( | mkldnn_primitive_desc_t * | reorder_primitive_desc, |
const_mkldnn_primitive_desc_t | input, | ||
const_mkldnn_primitive_desc_t | output, | ||
const_mkldnn_primitive_attr_t | attr | ||
) |
Initializes a reorder_primitive_desc
using an attr
attribute and descriptors of input
and output
memory primitives.
Order of inputs:
Order of outputs: