#include <mkldnn.hpp>
Inheritance diagram for mkldnn::batch_normalization_backward:
Collaboration diagram for mkldnn::batch_normalization_backward:
Classes | |
| struct | desc |
| struct | primitive_desc |
Public Member Functions | |
| batch_normalization_backward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &mean, const primitive::at &variance, const primitive::at &diff_dst, const primitive::at &weights, const memory &diff_src, const memory &diff_weights) | |
| batch_normalization_backward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &mean, const primitive::at &variance, const primitive::at &diff_dst, const primitive::at &weights, const primitive::at &workspace, const memory &diff_src, const memory &diff_weights) | |
| batch_normalization_backward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &mean, const primitive::at &variance, const primitive::at &diff_dst, const primitive::at &weights_or_workspace, const memory &diff_src) | |
| batch_normalization_backward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &mean, const primitive::at &variance, const primitive::at &diff_dst, const memory &diff_src) | |
 Public Member Functions inherited from mkldnn::primitive | |
| 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 (mkldnn_primitive_t t=0, bool weak=false) | |
| Constructs a C handle wrapper. More... | |
| 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... | |
Additional Inherited Members | |
| Public Types inherited from mkldnn::primitive | |
| 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::shuffle = mkldnn_shuffle, kind::eltwise = mkldnn_eltwise, 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::rnn = mkldnn_rnn } |
| A proxy to C primitive kind enum. More... | |
| Protected Member Functions inherited from mkldnn::handle< mkldnn_primitive_t > | |
| bool | operator== (const mkldnn_primitive_t other) const |
| bool | operator!= (const mkldnn_primitive_t other) const |
Constructor & Destructor Documentation
◆ batch_normalization_backward() [1/4]
|
inline |
◆ batch_normalization_backward() [2/4]
|
inline |
◆ batch_normalization_backward() [3/4]
|
inline |
- Warning
- This constructor works for backward_data propagation
- w/ weights but w/o workspace, or
- w/ workspace but w/o weights
◆ batch_normalization_backward() [4/4]
|
inline |
The documentation for this struct was generated from the following file:
- include/mkldnn.hpp
