|
| batch_normalization_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &mean, const primitive::at &variance, const primitive::at &weights, const memory &dst) |
|
| batch_normalization_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &mean, const primitive::at &variance, const memory &dst) |
|
| batch_normalization_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &weights, const memory &dst, const memory &mean, const memory &variance) |
|
| batch_normalization_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &weights, const memory &dst, const memory &mean, const memory &variance, const memory &workspace) |
|
| batch_normalization_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const memory &dst, const memory &mean, const memory &variance) |
|
| batch_normalization_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const memory &dst, const memory &mean, const memory &variance, const memory &workspace) |
|
| batch_normalization_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &weights, const memory &dst) |
|
| batch_normalization_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const memory &dst) |
|
const_mkldnn_primitive_desc_t | get_primitive_desc () const |
| Returns the descriptor of the underlying C API primitive. More...
|
|
Public Member Functions inherited from mkldnn::handle< mkldnn_primitive_t > |
| handle (const handle &other) |
|
handle & | operator= (const handle &other) |
|
bool | operator== (const handle &other) const |
|
bool | operator!= (const handle &other) const |
|
void | reset (mkldnn_primitive_t t, bool weak=false) |
| Resets the value of a C handle. More...
|
|
mkldnn_primitive_t | get () const |
| Returns the value of the underlying C handle. More...
|
|
|
enum | kind {
kind::undefined_primitive = mkldnn_undefined_primitive,
kind::memory = mkldnn_memory,
kind::view = mkldnn_view,
kind::reorder = mkldnn_reorder,
kind::concat = mkldnn_concat,
kind::concat_inplace = mkldnn_concat_inplace,
kind::sum = mkldnn_sum,
kind::convolution = mkldnn_convolution,
kind::deconvolution = mkldnn_deconvolution,
kind::eltwise = mkldnn_eltwise,
kind::relu = mkldnn_relu,
kind::softmax = mkldnn_softmax,
kind::pooling = mkldnn_pooling,
kind::lrn = mkldnn_lrn,
kind::batch_normalization = mkldnn_batch_normalization,
kind::inner_product = mkldnn_inner_product,
kind::convolution_relu = mkldnn_convolution_relu,
kind::rnn = mkldnn_rnn
} |
| A proxy to C primitive kind enum. More...
|
|
Protected Member Functions inherited from mkldnn::handle< mkldnn_primitive_t > |
| handle (mkldnn_primitive_t t=0, bool weak=false) |
| Constructs a C handle wrapper. More...
|
|
bool | operator== (const mkldnn_primitive_t other) const |
|
bool | operator!= (const mkldnn_primitive_t other) const |
|