| Package | Description |
|---|---|
| org.bytedeco.dnnl | |
| org.bytedeco.dnnl.global |
| Modifier and Type | Method and Description |
|---|---|
dnnl_primitive_attr |
dnnl_primitive_attr_handle.asDnnl_primitive_attr()
Converts a handle to the underlying C API handle type.
|
dnnl_primitive_attr |
dnnl_primitive_attr_handle.get() |
dnnl_primitive_attr |
dnnl_primitive_attr_handle.get(boolean allow_empty)
Returns the underlying C API handle.
|
| Modifier and Type | Method and Description |
|---|---|
void |
dnnl_primitive_attr_handle.reset(dnnl_primitive_attr t) |
void |
dnnl_primitive_attr_handle.reset(dnnl_primitive_attr t,
boolean weak)
Resets the handle wrapper objects to wrap a new C API handle.
|
| Constructor and Description |
|---|
dnnl_primitive_attr_handle(dnnl_primitive_attr t) |
dnnl_primitive_attr_handle(dnnl_primitive_attr t,
boolean weak)
Constructs a handle wrapper object from a C API handle.
|
primitive_attr(dnnl_primitive_attr t) |
primitive_attr(dnnl_primitive_attr t,
boolean weak) |
| Modifier and Type | Method and Description |
|---|---|
static int |
dnnl.dnnl_concat_primitive_desc_create(dnnl_primitive_desc concat_primitive_desc,
dnnl_memory_desc_t dst_desc,
int n,
int concat_dimension,
dnnl_memory_desc_t src_descs,
dnnl_primitive_attr attr,
dnnl_engine engine)
\} dnnl_api_reorder
|
static int |
dnnl.dnnl_concat_primitive_desc_create(PointerPointer concat_primitive_desc,
dnnl_memory_desc_t dst_desc,
int n,
int concat_dimension,
dnnl_memory_desc_t src_descs,
dnnl_primitive_attr attr,
dnnl_engine engine) |
static int |
dnnl.dnnl_primitive_attr_clone(dnnl_primitive_attr attr,
dnnl_primitive_attr existing_attr)
Clones primitive attributes.
|
static int |
dnnl.dnnl_primitive_attr_clone(PointerPointer attr,
dnnl_primitive_attr existing_attr) |
static int |
dnnl.dnnl_primitive_attr_create(dnnl_primitive_attr attr)
\} dnnl_api_primitives_common
|
static int |
dnnl.dnnl_primitive_attr_destroy(dnnl_primitive_attr attr)
Destroys primitive attributes.
|
static int |
dnnl.dnnl_primitive_attr_get_output_scales(dnnl_primitive_attr attr,
long[] count,
int[] mask,
float[] scales) |
static int |
dnnl.dnnl_primitive_attr_get_output_scales(dnnl_primitive_attr attr,
LongBuffer count,
IntBuffer mask,
FloatBuffer scales) |
static int |
dnnl.dnnl_primitive_attr_get_output_scales(dnnl_primitive_attr attr,
LongPointer count,
IntPointer mask,
FloatPointer scales) |
static int |
dnnl.dnnl_primitive_attr_get_output_scales(dnnl_primitive_attr attr,
LongPointer count,
IntPointer mask,
PointerPointer scales)
Returns primitive attributes output scaling factors correspondence mask
and values.
|
static int |
dnnl.dnnl_primitive_attr_get_post_ops(dnnl_primitive_attr attr,
dnnl_post_ops post_ops)
Returns primitive attributes post-ops.
|
static int |
dnnl.dnnl_primitive_attr_get_post_ops(dnnl_primitive_attr attr,
PointerPointer post_ops) |
static int |
dnnl.dnnl_primitive_attr_get_scales(dnnl_primitive_attr attr,
int arg,
long[] count,
int[] mask,
float[] scales) |
static int |
dnnl.dnnl_primitive_attr_get_scales(dnnl_primitive_attr attr,
int arg,
LongBuffer count,
IntBuffer mask,
FloatBuffer scales) |
static int |
dnnl.dnnl_primitive_attr_get_scales(dnnl_primitive_attr attr,
int arg,
LongPointer count,
IntPointer mask,
FloatPointer scales) |
static int |
dnnl.dnnl_primitive_attr_get_scales(dnnl_primitive_attr attr,
int arg,
LongPointer count,
IntPointer mask,
PointerPointer scales)
Returns primitive attributes scaling factors correspondence mask and values
for a given memory argument.
|
static int |
dnnl.dnnl_primitive_attr_get_scratchpad_mode(dnnl_primitive_attr attr,
int[] mode) |
static int |
dnnl.dnnl_primitive_attr_get_scratchpad_mode(dnnl_primitive_attr attr,
IntBuffer mode) |
static int |
dnnl.dnnl_primitive_attr_get_scratchpad_mode(dnnl_primitive_attr attr,
IntPointer mode)
Returns the primitive attributes scratchpad mode.
|
static int |
dnnl.dnnl_primitive_attr_get_zero_points(dnnl_primitive_attr attr,
int arg,
long[] count,
int[] mask,
int[] zero_points) |
static int |
dnnl.dnnl_primitive_attr_get_zero_points(dnnl_primitive_attr attr,
int arg,
LongBuffer count,
IntBuffer mask,
IntBuffer zero_points) |
static int |
dnnl.dnnl_primitive_attr_get_zero_points(dnnl_primitive_attr attr,
int arg,
LongPointer count,
IntPointer mask,
IntPointer zero_points) |
static int |
dnnl.dnnl_primitive_attr_get_zero_points(dnnl_primitive_attr attr,
int arg,
LongPointer count,
IntPointer mask,
PointerPointer zero_points)
Returns \p count, correspondence zero point \p mask, and a pointer to a
constant int32_t array of \p zero_points for given \p attr and memory
argument (index), previously set by dnnl_primitive_attr_set_zero_points.
|
static int |
dnnl.dnnl_primitive_attr_set_output_scales(dnnl_primitive_attr attr,
long count,
int mask,
float[] scales) |
static int |
dnnl.dnnl_primitive_attr_set_output_scales(dnnl_primitive_attr attr,
long count,
int mask,
FloatBuffer scales) |
static int |
dnnl.dnnl_primitive_attr_set_output_scales(dnnl_primitive_attr attr,
long count,
int mask,
FloatPointer scales)
Sets output scaling factors correspondence mask and values.
|
static int |
dnnl.dnnl_primitive_attr_set_post_ops(dnnl_primitive_attr attr,
dnnl_post_ops post_ops)
Sets primitive attributes post-ops.
|
static int |
dnnl.dnnl_primitive_attr_set_rnn_data_qparams(dnnl_primitive_attr attr,
float scale,
float shift)
\} dnnl_api_inner_product
|
static int |
dnnl.dnnl_primitive_attr_set_rnn_weights_qparams(dnnl_primitive_attr attr,
long count,
int mask,
float[] scales) |
static int |
dnnl.dnnl_primitive_attr_set_rnn_weights_qparams(dnnl_primitive_attr attr,
long count,
int mask,
FloatBuffer scales) |
static int |
dnnl.dnnl_primitive_attr_set_rnn_weights_qparams(dnnl_primitive_attr attr,
long count,
int mask,
FloatPointer scales)
Sets quantization scaling factors for RNN weights tensors.
|
static int |
dnnl.dnnl_primitive_attr_set_scales(dnnl_primitive_attr attr,
int arg,
long count,
int mask,
float[] scales) |
static int |
dnnl.dnnl_primitive_attr_set_scales(dnnl_primitive_attr attr,
int arg,
long count,
int mask,
FloatBuffer scales) |
static int |
dnnl.dnnl_primitive_attr_set_scales(dnnl_primitive_attr attr,
int arg,
long count,
int mask,
FloatPointer scales)
Sets primitive attributes scaling factors for primitive operations for a
given memory argument.
|
static int |
dnnl.dnnl_primitive_attr_set_scratchpad_mode(dnnl_primitive_attr attr,
int mode)
Sets primitive attributes scratchpad mode.
|
static int |
dnnl.dnnl_primitive_attr_set_zero_points(dnnl_primitive_attr attr,
int arg,
long count,
int mask,
int[] zero_points) |
static int |
dnnl.dnnl_primitive_attr_set_zero_points(dnnl_primitive_attr attr,
int arg,
long count,
int mask,
IntBuffer zero_points) |
static int |
dnnl.dnnl_primitive_attr_set_zero_points(dnnl_primitive_attr attr,
int arg,
long count,
int mask,
IntPointer zero_points)
Sets primitive attributes zero points for primitive operations for a given
memory argument.
|
static int |
dnnl.dnnl_primitive_desc_create(dnnl_primitive_desc primitive_desc,
const_dnnl_op_desc_t op_desc,
dnnl_primitive_attr attr,
dnnl_engine engine,
dnnl_primitive_desc hint_forward_primitive_desc)
Creates a primitive descriptor.
|
static int |
dnnl.dnnl_primitive_desc_create(PointerPointer primitive_desc,
const_dnnl_op_desc_t op_desc,
dnnl_primitive_attr attr,
dnnl_engine engine,
dnnl_primitive_desc hint_forward_primitive_desc) |
static int |
dnnl.dnnl_primitive_desc_get_attr(dnnl_primitive_desc primitive_desc,
dnnl_primitive_attr attr)
Returns a constant reference to the attributes of a primitive descriptor.
|
static int |
dnnl.dnnl_primitive_desc_iterator_create(dnnl_primitive_desc_iterator iterator,
const_dnnl_op_desc_t op_desc,
dnnl_primitive_attr attr,
dnnl_engine engine,
dnnl_primitive_desc hint_forward_primitive_desc)
\addtogroup dnnl_api
\{
|
static int |
dnnl.dnnl_primitive_desc_iterator_create(PointerPointer iterator,
const_dnnl_op_desc_t op_desc,
dnnl_primitive_attr attr,
dnnl_engine engine,
dnnl_primitive_desc hint_forward_primitive_desc) |
static int |
dnnl.dnnl_reorder_primitive_desc_create(dnnl_primitive_desc reorder_primitive_desc,
dnnl_memory_desc_t src_desc,
dnnl_engine src_engine,
dnnl_memory_desc_t dst_desc,
dnnl_engine dst_engine,
dnnl_primitive_attr attr)
\} dnnl_api_memory
|
static int |
dnnl.dnnl_reorder_primitive_desc_create(PointerPointer reorder_primitive_desc,
dnnl_memory_desc_t src_desc,
dnnl_engine src_engine,
dnnl_memory_desc_t dst_desc,
dnnl_engine dst_engine,
dnnl_primitive_attr attr) |
static int |
dnnl.dnnl_sum_primitive_desc_create(dnnl_primitive_desc sum_primitive_desc,
dnnl_memory_desc_t dst_desc,
int n,
float[] scales,
dnnl_memory_desc_t src_descs,
dnnl_primitive_attr attr,
dnnl_engine engine) |
static int |
dnnl.dnnl_sum_primitive_desc_create(dnnl_primitive_desc sum_primitive_desc,
dnnl_memory_desc_t dst_desc,
int n,
FloatBuffer scales,
dnnl_memory_desc_t src_descs,
dnnl_primitive_attr attr,
dnnl_engine engine) |
static int |
dnnl.dnnl_sum_primitive_desc_create(dnnl_primitive_desc sum_primitive_desc,
dnnl_memory_desc_t dst_desc,
int n,
FloatPointer scales,
dnnl_memory_desc_t src_descs,
dnnl_primitive_attr attr,
dnnl_engine engine)
\} dnnl_api_concat
|
static int |
dnnl.dnnl_sum_primitive_desc_create(PointerPointer sum_primitive_desc,
dnnl_memory_desc_t dst_desc,
int n,
float[] scales,
dnnl_memory_desc_t src_descs,
dnnl_primitive_attr attr,
dnnl_engine engine) |
static int |
dnnl.dnnl_sum_primitive_desc_create(PointerPointer sum_primitive_desc,
dnnl_memory_desc_t dst_desc,
int n,
FloatBuffer scales,
dnnl_memory_desc_t src_descs,
dnnl_primitive_attr attr,
dnnl_engine engine) |
static int |
dnnl.dnnl_sum_primitive_desc_create(PointerPointer sum_primitive_desc,
dnnl_memory_desc_t dst_desc,
int n,
FloatPointer scales,
dnnl_memory_desc_t src_descs,
dnnl_primitive_attr attr,
dnnl_engine engine) |
Copyright © 2020. All rights reserved.