#include <mkldnn.hpp>
Inheritance diagram for mkldnn::rnn_forward:
Collaboration diagram for mkldnn::rnn_forward:
Classes | |
| struct | desc |
| struct | primitive_desc |
Public Member Functions | |
| rnn_forward (const primitive_desc &aprimitive_desc, const primitive::at &src_layer, const primitive::at &src_iter, const primitive::at &weights_layer, const primitive::at &weights_iter, const primitive::at &bias, const memory &dst_layer, const memory &dst_iter, const memory &workspace) | |
 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
◆ rnn_forward()
|
inline |
The documentation for this struct was generated from the following file:
- include/mkldnn.hpp
