Package | Description |
---|---|
org.bytedeco.mkldnn | |
org.bytedeco.mkldnn.global |
Modifier and Type | Method and Description |
---|---|
mkldnn_primitive_attr |
mkldnn_primitive_attr_handle.get()
Returns the value of the underlying C handle.
|
Modifier and Type | Method and Description |
---|---|
void |
mkldnn_primitive_attr_handle.reset(mkldnn_primitive_attr t) |
void |
mkldnn_primitive_attr_handle.reset(mkldnn_primitive_attr t,
boolean weak)
Resets the value of a C handle.
|
Constructor and Description |
---|
mkldnn_primitive_attr_handle(mkldnn_primitive_attr t,
boolean weak)
Constructs a C handle wrapper.
|
Modifier and Type | Method and Description |
---|---|
static int |
mkldnn.mkldnn_primitive_attr_clone(mkldnn_primitive_attr attr,
mkldnn_primitive_attr existing_attr)
Makes a copy of an \p existing_attr.
|
static int |
mkldnn.mkldnn_primitive_attr_clone(PointerPointer attr,
mkldnn_primitive_attr existing_attr) |
static int |
mkldnn.mkldnn_primitive_attr_create(mkldnn_primitive_attr attr)
Creates an empty (default) \p attr attribute.
|
static int |
mkldnn.mkldnn_primitive_attr_destroy(mkldnn_primitive_attr attr)
Deletes an \p attr.
|
static int |
mkldnn.mkldnn_primitive_attr_get_int_output_round_mode(mkldnn_primitive_attr attr,
int[] round_mode) |
static int |
mkldnn.mkldnn_primitive_attr_get_int_output_round_mode(mkldnn_primitive_attr attr,
IntBuffer round_mode) |
static int |
mkldnn.mkldnn_primitive_attr_get_int_output_round_mode(mkldnn_primitive_attr attr,
IntPointer round_mode)
Returns integer output rounding mode \p round_mode for a given \p attr,
previously set by mkldnn_primitive_attr_set_int_output_round_mode.
|
static int |
mkldnn.mkldnn_primitive_attr_get_output_scales(mkldnn_primitive_attr attr,
int[] count,
int[] mask,
float[] scales) |
static int |
mkldnn.mkldnn_primitive_attr_get_output_scales(mkldnn_primitive_attr attr,
IntBuffer count,
IntBuffer mask,
FloatBuffer scales) |
static int |
mkldnn.mkldnn_primitive_attr_get_output_scales(mkldnn_primitive_attr attr,
IntPointer count,
IntPointer mask,
FloatPointer scales) |
static int |
mkldnn.mkldnn_primitive_attr_get_output_scales(mkldnn_primitive_attr attr,
IntPointer count,
IntPointer mask,
PointerPointer scales)
Returns \p count, correspondence scale \p mask, and a pointer to a constant
floating point array of output \p scales for given \p attr, previously set
by mkldnn_primitive_attr_set_output_scales.
|
static int |
mkldnn.mkldnn_primitive_attr_get_post_ops(mkldnn_primitive_attr attr,
mkldnn_post_ops post_ops)
Returns \p post_ops for given \p attr.
|
static int |
mkldnn.mkldnn_primitive_attr_get_post_ops(mkldnn_primitive_attr attr,
PointerPointer post_ops) |
static int |
mkldnn.mkldnn_primitive_attr_set_int_output_round_mode(mkldnn_primitive_attr attr,
int round_mode)
Sets output rounding mode \p round_mode for integer operations for a given
\p attr.
|
static int |
mkldnn.mkldnn_primitive_attr_set_output_scales(mkldnn_primitive_attr attr,
int count,
int mask,
float[] scales) |
static int |
mkldnn.mkldnn_primitive_attr_set_output_scales(mkldnn_primitive_attr attr,
int count,
int mask,
FloatBuffer scales) |
static int |
mkldnn.mkldnn_primitive_attr_set_output_scales(mkldnn_primitive_attr attr,
int count,
int mask,
FloatPointer scales)
Sets output \p scales for primitive operations.
|
static int |
mkldnn.mkldnn_primitive_attr_set_post_ops(mkldnn_primitive_attr attr,
mkldnn_post_ops post_ops)
Sets configured \p post_ops to an attribute \p attr for future use (when
primitive descriptor is being created).
|
static int |
mkldnn.mkldnn_primitive_attr_set_rnn_data_qparams(mkldnn_primitive_attr attr,
float scale,
float shift)
Sets quantization \p scale and \p shift for RNN data tensors.
|
static int |
mkldnn.mkldnn_primitive_attr_set_rnn_weights_qparams(mkldnn_primitive_attr attr,
int count,
int mask,
float[] weights_scales) |
static int |
mkldnn.mkldnn_primitive_attr_set_rnn_weights_qparams(mkldnn_primitive_attr attr,
int count,
int mask,
FloatBuffer weights_scales) |
static int |
mkldnn.mkldnn_primitive_attr_set_rnn_weights_qparams(mkldnn_primitive_attr attr,
int count,
int mask,
FloatPointer weights_scales)
Sets quantization scales \p weights_scales for RNN weights tensors.
|
static int |
mkldnn.mkldnn_primitive_desc_create_v2(mkldnn_primitive_desc primitive_desc,
const_mkldnn_op_desc_t op_desc,
mkldnn_primitive_attr attr,
mkldnn_engine engine,
mkldnn_primitive_desc hint_forward_primitive_desc)
Creates a \p primitive_desc using \p op_desc, \p attr, \p engine, and
optionally a hint primitive descriptor from forward propagation.
|
static int |
mkldnn.mkldnn_primitive_desc_create_v2(PointerPointer primitive_desc,
const_mkldnn_op_desc_t op_desc,
mkldnn_primitive_attr attr,
mkldnn_engine engine,
mkldnn_primitive_desc hint_forward_primitive_desc) |
static int |
mkldnn.mkldnn_primitive_desc_get_attr(mkldnn_primitive_desc primitive_desc,
mkldnn_primitive_attr attr)
Returns a constant reference to the attribute of a \p primitive_desc.
|
static int |
mkldnn.mkldnn_primitive_desc_iterator_create_v2(mkldnn_primitive_desc_iterator iterator,
const_mkldnn_op_desc_t op_desc,
mkldnn_primitive_attr attr,
mkldnn_engine engine,
mkldnn_primitive_desc hint_forward_primitive_desc)
Creates a primitive descriptor \p iterator for given \p op_desc, \p attr,
\p engine, and optionally a hint primitive descriptor from forward
propagation (required for backward propagation).
|
static int |
mkldnn.mkldnn_primitive_desc_iterator_create_v2(PointerPointer iterator,
const_mkldnn_op_desc_t op_desc,
mkldnn_primitive_attr attr,
mkldnn_engine engine,
mkldnn_primitive_desc hint_forward_primitive_desc) |
static int |
mkldnn.mkldnn_reorder_primitive_desc_create_v2(mkldnn_primitive_desc reorder_primitive_desc,
mkldnn_primitive_desc input,
mkldnn_primitive_desc output,
mkldnn_primitive_attr attr)
Initializes a \p reorder_primitive_desc using an \p attr attribute and
descriptors of \p input and \p output memory primitives.
|
static int |
mkldnn.mkldnn_reorder_primitive_desc_create_v2(PointerPointer reorder_primitive_desc,
mkldnn_primitive_desc input,
mkldnn_primitive_desc output,
mkldnn_primitive_attr attr) |
Copyright © 2022. All rights reserved.