Intel(R) Math Kernel Library for Deep Neural Networks (Intel(R) MKL-DNN)
0.21.0
Performance library for Deep Learning
|
#include <mkldnn.hpp>
Public Member Functions | |
primitive_desc (const desc &desc, const engine &e, const rnn_forward::primitive_desc &hint_fwd_pd) | |
primitive_desc (const desc &desc, const primitive_attr &attr, const engine &e, const rnn_forward::primitive_desc &hint_fwd_pd) | |
REG_QUERY_MPD (src_layer, src, 0) | |
REG_QUERY_MPD (src_iter, src, 1) | |
REG_QUERY_MPD (weights_layer, weights, 0) | |
REG_QUERY_MPD (weights_iter, weights, 1) | |
REG_QUERY_MPD (bias, weights, 2) | |
REG_QUERY_MPD (dst_layer, dst, 0) | |
REG_QUERY_MPD (dst_iter, dst, 1) | |
REG_QUERY_MPD (workspace, workspace, 0) | |
REG_QUERY_MPD (diff_src_layer, diff_src, 0) | |
REG_QUERY_MPD (diff_src_iter, diff_src, 1) | |
REG_QUERY_MPD (diff_weights_layer, diff_weights, 0) | |
REG_QUERY_MPD (diff_weights_iter, diff_weights, 1) | |
REG_QUERY_MPD (diff_bias, diff_weights, 2) | |
REG_QUERY_MPD (diff_dst_layer, diff_dst, 0) | |
REG_QUERY_MPD (diff_dst_iter, diff_dst, 1) | |
Public Member Functions inherited from mkldnn::primitive_desc | |
primitive_desc (const_mkldnn_op_desc_t desc, const primitive_attr *attr, const engine &e, const_mkldnn_primitive_desc_t hint_fwd_pd) | |
engine | get_engine () |
primitive_attr | get_primitive_attr () const |
const char * | impl_info_str () const |
Returns implementation name. More... | |
bool | next_impl () |
Advances the next implementation for the given op descriptor. More... | |
memory::primitive_desc | query_mpd (query what, int idx=0) const |
Queries and returns requested memory primitive descriptor. More... | |
Public Member Functions inherited from mkldnn::handle< mkldnn_primitive_desc_t > | |
handle (mkldnn_primitive_desc_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_desc_t t, bool weak=false) |
Resets the value of a C handle. More... | |
mkldnn_primitive_desc_t | get () const |
Returns the value of the underlying C handle. More... | |
Additional Inherited Members | |
Protected Member Functions inherited from mkldnn::handle< mkldnn_primitive_desc_t > | |
bool | operator== (const mkldnn_primitive_desc_t other) const |
bool | operator!= (const mkldnn_primitive_desc_t other) const |
|
inline |
|
inline |
mkldnn::rnn_backward::primitive_desc::REG_QUERY_MPD | ( | src_layer | , |
src | , | ||
0 | |||
) |
mkldnn::rnn_backward::primitive_desc::REG_QUERY_MPD | ( | src_iter | , |
src | , | ||
1 | |||
) |
mkldnn::rnn_backward::primitive_desc::REG_QUERY_MPD | ( | weights_layer | , |
weights | , | ||
0 | |||
) |
mkldnn::rnn_backward::primitive_desc::REG_QUERY_MPD | ( | weights_iter | , |
weights | , | ||
1 | |||
) |
mkldnn::rnn_backward::primitive_desc::REG_QUERY_MPD | ( | bias | , |
weights | , | ||
2 | |||
) |
mkldnn::rnn_backward::primitive_desc::REG_QUERY_MPD | ( | dst_layer | , |
dst | , | ||
0 | |||
) |
mkldnn::rnn_backward::primitive_desc::REG_QUERY_MPD | ( | dst_iter | , |
dst | , | ||
1 | |||
) |
mkldnn::rnn_backward::primitive_desc::REG_QUERY_MPD | ( | workspace | , |
workspace | , | ||
0 | |||
) |
mkldnn::rnn_backward::primitive_desc::REG_QUERY_MPD | ( | diff_src_layer | , |
diff_src | , | ||
0 | |||
) |
mkldnn::rnn_backward::primitive_desc::REG_QUERY_MPD | ( | diff_src_iter | , |
diff_src | , | ||
1 | |||
) |
mkldnn::rnn_backward::primitive_desc::REG_QUERY_MPD | ( | diff_weights_layer | , |
diff_weights | , | ||
0 | |||
) |
mkldnn::rnn_backward::primitive_desc::REG_QUERY_MPD | ( | diff_weights_iter | , |
diff_weights | , | ||
1 | |||
) |
mkldnn::rnn_backward::primitive_desc::REG_QUERY_MPD | ( | diff_bias | , |
diff_weights | , | ||
2 | |||
) |
mkldnn::rnn_backward::primitive_desc::REG_QUERY_MPD | ( | diff_dst_layer | , |
diff_dst | , | ||
0 | |||
) |
mkldnn::rnn_backward::primitive_desc::REG_QUERY_MPD | ( | diff_dst_iter | , |
diff_dst | , | ||
1 | |||
) |