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 | |
desc (prop_kind aprop_kind, const memory::desc &data_desc, int softmax_axis) | |
Public Attributes | |
mkldnn_softmax_desc_t | data |
|
inline |
mkldnn_softmax_desc_t mkldnn::softmax_forward::desc::data |