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_engine_create(dnnl_engine engine,
int kind,
long index)
Creates an engine.
|
static int |
dnnl.dnnl_engine_destroy(dnnl_engine engine)
Destroys an engine.
|
static int |
dnnl.dnnl_engine_get_kind(dnnl_engine engine,
int[] kind) |
static int |
dnnl.dnnl_engine_get_kind(dnnl_engine engine,
IntBuffer kind) |
static int |
dnnl.dnnl_engine_get_kind(dnnl_engine engine,
IntPointer kind)
Returns the kind of an engine.
|
static int |
dnnl.dnnl_memory_create(dnnl_memory memory,
dnnl_memory_desc_t memory_desc,
dnnl_engine engine,
Pointer handle)
Creates a memory object.
|
static int |
dnnl.dnnl_memory_create(PointerPointer memory,
dnnl_memory_desc_t memory_desc,
dnnl_engine engine,
Pointer handle) |
static int |
dnnl.dnnl_memory_get_engine(dnnl_memory memory,
dnnl_engine engine)
Returns the engine of a memory object.
|
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_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_stream_create_v2(dnnl_stream stream,
dnnl_engine engine,
int flags,
dnnl_stream_attr attr)
Creates an execution stream.
|
static int |
dnnl.dnnl_stream_create_v2(PointerPointer stream,
dnnl_engine engine,
int flags,
dnnl_stream_attr attr) |
static int |
dnnl.dnnl_stream_create(dnnl_stream stream,
dnnl_engine engine,
int flags)
Creates an execution stream.
|
static int |
dnnl.dnnl_stream_create(PointerPointer stream,
dnnl_engine engine,
int flags) |
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) |