Package | Description |
---|---|
org.bytedeco.mkldnn.global |
Modifier and Type | Method and Description |
---|---|
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(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_destroy(mkldnn_primitive_desc_iterator iterator)
Deletes a primitive descriptor \p iterator
|
static mkldnn_primitive_desc |
mkldnn.mkldnn_primitive_desc_iterator_fetch(mkldnn_primitive_desc_iterator iterator)
Fetches the current primitive descriptor.
|
static int |
mkldnn.mkldnn_primitive_desc_iterator_next(mkldnn_primitive_desc_iterator iterator)
Iterates over primitive descriptors.
|
Copyright © 2022. All rights reserved.