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 | |
std::vector< const_mkldnn_primitive_desc_t > | cpp_to_c (std::vector< memory::primitive_desc > inputs) |
primitive_desc (const memory::desc &output, const std::vector< float > &scales, std::vector< memory::primitive_desc > inputs) | |
primitive_desc (const std::vector< float > &scales, std::vector< memory::primitive_desc > inputs) | |
memory::primitive_desc | dst_primitive_desc () const |
engine | get_engine () |
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 |
|
inline |
|
inline |
|
inline |