Package | Description |
---|---|
org.bytedeco.mkldnn | |
org.bytedeco.mkldnn.global |
Modifier and Type | Method and Description |
---|---|
mkldnn_primitive_desc |
primitive.get_mkldnn_primitive_desc()
Returns the descriptor of the underlying C API primitive.
|
mkldnn_primitive_desc |
mkldnn_primitive_desc_handle.get()
Returns the value of the underlying C handle.
|
mkldnn_primitive_desc[] |
mkldnn_primitive_desc_vector.get() |
mkldnn_primitive_desc |
mkldnn_primitive_desc_vector.Iterator.get() |
mkldnn_primitive_desc |
mkldnn_primitive_desc_vector.get(long i) |
mkldnn_primitive_desc |
mkldnn_primitive_desc_vector.pop_back() |
Modifier and Type | Method and Description |
---|---|
mkldnn_primitive_desc_vector.Iterator |
mkldnn_primitive_desc_vector.insert(mkldnn_primitive_desc_vector.Iterator pos,
mkldnn_primitive_desc value) |
mkldnn_primitive_desc_vector |
mkldnn_primitive_desc_vector.push_back(mkldnn_primitive_desc value) |
mkldnn_primitive_desc_vector |
mkldnn_primitive_desc_vector.put(long i,
mkldnn_primitive_desc value) |
mkldnn_primitive_desc_vector |
mkldnn_primitive_desc_vector.put(mkldnn_primitive_desc... array) |
mkldnn_primitive_desc_vector |
mkldnn_primitive_desc_vector.put(mkldnn_primitive_desc value) |
void |
mkldnn_primitive_desc_handle.reset(mkldnn_primitive_desc t) |
void |
mkldnn_primitive_desc_handle.reset(mkldnn_primitive_desc t,
boolean weak)
Resets the value of a C handle.
|
Constructor and Description |
---|
mkldnn_primitive_desc_handle(mkldnn_primitive_desc t,
boolean weak)
Constructs a C handle wrapper.
|
mkldnn_primitive_desc_vector(mkldnn_primitive_desc... array) |
mkldnn_primitive_desc_vector(mkldnn_primitive_desc value) |
primitive_desc(const_mkldnn_op_desc_t desc,
primitive_attr attr,
engine e,
mkldnn_primitive_desc hint_fwd_pd) |
Modifier and Type | Method and Description |
---|---|
static mkldnn_primitive_desc |
mkldnn.mkldnn_primitive_desc_iterator_fetch(mkldnn_primitive_desc_iterator iterator)
Fetches the current primitive descriptor.
|
static mkldnn_primitive_desc |
mkldnn.mkldnn_primitive_desc_query_pd(mkldnn_primitive_desc primitive_desc,
int what,
int index)
Queries primitive descriptor for primitive descriptor
|
Modifier and Type | Method and Description |
---|---|
static void |
mkldnn.check_num_parameters(mkldnn_primitive_desc aprimitive_desc,
int n_inputs,
int n_outputs,
BytePointer prim_name) |
static void |
mkldnn.check_num_parameters(mkldnn_primitive_desc aprimitive_desc,
int n_inputs,
int n_outputs,
String prim_name) |
static int |
mkldnn.mkldnn_concat_primitive_desc_create(mkldnn_primitive_desc concat_primitive_desc,
mkldnn_memory_desc_t output_desc,
int n,
int concat_dimension,
mkldnn_primitive_desc input_pds)
Creates out-of-place \p concat_primitive_desc for concatenation of \p n
inputs by \p concat_dimension with resulting \p output_desc memory
descriptor.
|
static int |
mkldnn.mkldnn_memory_primitive_desc_create(mkldnn_primitive_desc memory_primitive_desc,
mkldnn_memory_desc_t memory_desc,
mkldnn_engine engine)
Creates a \p memory_primitive_desc memory primitive descriptor using \p
memory_desc and \p engine.
|
static int |
mkldnn.mkldnn_memory_primitive_desc_equal(mkldnn_primitive_desc lhs,
mkldnn_primitive_desc rhs)
Compares two descriptors of memory primitives.
|
static long |
mkldnn.mkldnn_memory_primitive_desc_get_size(mkldnn_primitive_desc memory_primitive_desc)
Returns the size (in bytes) that is required for given \p
memory_primitive_desc
|
static int |
mkldnn.mkldnn_primitive_create(mkldnn_primitive primitive,
mkldnn_primitive_desc primitive_desc,
mkldnn_primitive_at_t inputs,
mkldnn_primitive outputs)
Creates a \p primitive using a \p primitive_desc descriptor and arrays of
\p inputs and \p outputs.
|
static int |
mkldnn.mkldnn_primitive_create(PointerPointer primitive,
mkldnn_primitive_desc primitive_desc,
mkldnn_primitive_at_t inputs,
PointerPointer outputs) |
static int |
mkldnn.mkldnn_primitive_desc_clone(mkldnn_primitive_desc primitive_desc,
mkldnn_primitive_desc existing_primitive_desc)
Makes a copy of a \p primitive_desc.
|
static int |
mkldnn.mkldnn_primitive_desc_clone(PointerPointer primitive_desc,
mkldnn_primitive_desc existing_primitive_desc) |
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_create(mkldnn_primitive_desc primitive_desc,
const_mkldnn_op_desc_t op_desc,
mkldnn_engine engine,
mkldnn_primitive_desc hint_forward_primitive_desc)
Creates a \p primitive_desc using \p op_desc, \p engine, and optionally a
hint primitive descriptor from forward propagation.
|
static int |
mkldnn.mkldnn_primitive_desc_create(PointerPointer primitive_desc,
const_mkldnn_op_desc_t op_desc,
mkldnn_engine engine,
mkldnn_primitive_desc hint_forward_primitive_desc) |
static int |
mkldnn.mkldnn_primitive_desc_destroy(mkldnn_primitive_desc primitive_desc)
Deletes a \p 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_get_attr(mkldnn_primitive_desc primitive_desc,
PointerPointer attr) |
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_primitive_desc_iterator_create(mkldnn_primitive_desc_iterator iterator,
const_mkldnn_op_desc_t op_desc,
mkldnn_engine engine,
mkldnn_primitive_desc hint_forward_primitive_desc)
Creates a primitive descriptor \p iterator for given \p op_desc, \p engine,
and optionally a hint primitive descriptor from forward propagation
(required for backward propagation).
|
static int |
mkldnn.mkldnn_primitive_desc_iterator_create(PointerPointer iterator,
const_mkldnn_op_desc_t op_desc,
mkldnn_engine engine,
mkldnn_primitive_desc hint_forward_primitive_desc) |
static mkldnn_memory_desc_t |
mkldnn.mkldnn_primitive_desc_query_memory_d(mkldnn_primitive_desc primitive_desc)
Queries primitive descriptor for memory descriptor
|
static mkldnn_primitive_desc |
mkldnn.mkldnn_primitive_desc_query_pd(mkldnn_primitive_desc primitive_desc,
int what,
int index)
Queries primitive descriptor for primitive descriptor
|
static int |
mkldnn.mkldnn_primitive_desc_query_s32(mkldnn_primitive_desc primitive_desc,
int what,
int index)
Queries primitive descriptor for signed 32bit int
|
static int |
mkldnn.mkldnn_primitive_desc_query(mkldnn_primitive_desc primitive_desc,
int what,
int index,
Pointer result)
Queries primitive descriptor
One of the most typical use cases is to query a convolution primitive
descriptor created with source, weights, and destination formats equal
to #mkldnn_any about the corresponding memory primitive descriptors
(\p what equals #mkldnn_query_src_pd, #mkldnn_query_weights_pd, and
#mkldnn_query_dst_pd respectively) to be able to prepare memory and
create reorders if required.
|
static int |
mkldnn.mkldnn_primitive_get_primitive_desc(mkldnn_primitive primitive,
mkldnn_primitive_desc primitive_desc)
Retrieves a reference to the \p primitive_desc descriptor of given \p
primitive.
|
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) |
static int |
mkldnn.mkldnn_reorder_primitive_desc_create(mkldnn_primitive_desc reorder_primitive_desc,
mkldnn_primitive_desc input,
mkldnn_primitive_desc output)
Initializes a \p reorder_primitive_desc using descriptors of \p input and
\p output memory primitives.
|
static int |
mkldnn.mkldnn_reorder_primitive_desc_create(PointerPointer reorder_primitive_desc,
mkldnn_primitive_desc input,
mkldnn_primitive_desc output) |
static int |
mkldnn.mkldnn_sum_primitive_desc_create(mkldnn_primitive_desc sum_primitive_desc,
mkldnn_memory_desc_t output_desc,
int n,
float[] scales,
mkldnn_primitive_desc input_pds) |
static int |
mkldnn.mkldnn_sum_primitive_desc_create(mkldnn_primitive_desc sum_primitive_desc,
mkldnn_memory_desc_t output_desc,
int n,
FloatBuffer scales,
mkldnn_primitive_desc input_pds) |
static int |
mkldnn.mkldnn_sum_primitive_desc_create(mkldnn_primitive_desc sum_primitive_desc,
mkldnn_memory_desc_t output_desc,
int n,
FloatPointer scales,
mkldnn_primitive_desc input_pds)
Creates out-of-place \p sum_primitive_desc for sum of \p n
inputs multiplied by scale with resulting \p output_desc memory
descriptor.
|
static int |
mkldnn.mkldnn_view_primitive_desc_create(mkldnn_primitive_desc view_primitive_desc,
mkldnn_primitive_desc memory_primitive_desc,
int[] dims,
int[] offsets) |
static int |
mkldnn.mkldnn_view_primitive_desc_create(mkldnn_primitive_desc view_primitive_desc,
mkldnn_primitive_desc memory_primitive_desc,
IntBuffer dims,
IntBuffer offsets) |
static int |
mkldnn.mkldnn_view_primitive_desc_create(mkldnn_primitive_desc view_primitive_desc,
mkldnn_primitive_desc memory_primitive_desc,
IntPointer dims,
IntPointer offsets)
Creates a \p view_primitive_desc for a given \p memory_primitive_desc, with
\p dims sizes and \p offsets offsets.
|
static int |
mkldnn.mkldnn_view_primitive_desc_create(PointerPointer view_primitive_desc,
mkldnn_primitive_desc memory_primitive_desc,
int[] dims,
int[] offsets) |
static int |
mkldnn.mkldnn_view_primitive_desc_create(PointerPointer view_primitive_desc,
mkldnn_primitive_desc memory_primitive_desc,
IntBuffer dims,
IntBuffer offsets) |
static int |
mkldnn.mkldnn_view_primitive_desc_create(PointerPointer view_primitive_desc,
mkldnn_primitive_desc memory_primitive_desc,
IntPointer dims,
IntPointer offsets) |
Copyright © 2022. All rights reserved.