|
| mkldnn::handle< T, traits >::handle (T t=0, bool weak=false) |
| Constructs a C handle wrapper. More...
|
|
bool | mkldnn::handle< T, traits >::operator== (const T other) const |
|
bool | mkldnn::handle< T, traits >::operator!= (const T other) const |
|
| mkldnn::handle< T, traits >::handle (const handle &other) |
|
handle & | mkldnn::handle< T, traits >::operator= (const handle &other) |
|
void | mkldnn::handle< T, traits >::reset (T t, bool weak=false) |
| Resets the value of a C handle. More...
|
|
T | mkldnn::handle< T, traits >::get () const |
| Returns the value of the underlying C handle. More...
|
|
bool | mkldnn::handle< T, traits >::operator== (const handle &other) const |
|
bool | mkldnn::handle< T, traits >::operator!= (const handle &other) const |
|
| mkldnn::primitive::at::at (const primitive &aprimitive, size_t at=0) |
| Constructs a wrapper specifying aprimitive output with index at . More...
|
|
| mkldnn::primitive::at::operator primitive () const |
| Returns the specified output. More...
|
|
const_mkldnn_primitive_desc_t | mkldnn::primitive::get_primitive_desc () const |
| Returns the descriptor of the underlying C API primitive. More...
|
|
| mkldnn::error::error (mkldnn_status_t astatus, std::string amessage, mkldnn_primitive_t aerror_primitive=0) |
| Constructs an error instance. More...
|
|
static void | mkldnn::error::wrap_c_api (mkldnn_status_t status, std::string message, mkldnn_primitive_t *error_primitive=0) |
| A convenience function for wrapping calls to the C API. Checks the return status and throws an error in case of failure. More...
|
|
static size_t | mkldnn::engine::get_count (kind akind) |
| Returns the number of engines of a certain kind. More...
|
|
| mkldnn::engine::engine (kind akind, size_t index) |
| Constructs an engine. More...
|
|
| mkldnn::engine::engine (const mkldnn_engine_t &aengine) |
|
| mkldnn::engine::engine (const handle< mkldnn_primitive_desc_t > &pd) |
|
template<class primitive_desc > |
static engine | mkldnn::engine::query (const primitive_desc &pd) |
|
template<typename T > |
static void | mkldnn::memory::validate_dims (std::vector< T > v) |
|
| mkldnn::memory::desc::desc (dims adims, data_type adata_type, format aformat) |
| Constructs a memory descriptor. More...
|
|
| mkldnn::memory::desc::desc (const mkldnn_memory_desc_t &adata) |
| Constructs a memory descriptor from a C API data structure. More...
|
|
| mkldnn::memory::primitive_desc::primitive_desc () |
|
| mkldnn::memory::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine) |
| Constructs a memory primitive descriptor. More...
|
|
memory::desc | mkldnn::memory::primitive_desc::desc () |
| Returns the memory primitive descriptor. More...
|
|
size_t | mkldnn::memory::primitive_desc::get_size () const |
| Returns the number of data elements in the memory described. More...
|
|
bool | mkldnn::memory::primitive_desc::operator== (const primitive_desc &other) const |
|
bool | mkldnn::memory::primitive_desc::operator!= (const primitive_desc &other) const |
|
engine | mkldnn::memory::primitive_desc::get_engine () |
|
| mkldnn::memory::memory (const primitive &aprimitive) |
| Constructs a memory primitive from a generic primitive. More...
|
|
| mkldnn::memory::memory (const primitive_desc &adesc) |
| Constructs a memory primitive. More...
|
|
| mkldnn::memory::memory (const primitive_desc &adesc, void *ahandle) |
|
primitive_desc | mkldnn::memory::get_primitive_desc () const |
| Returns the descriptor of the memory primitive. More...
|
|
void * | mkldnn::memory::get_data_handle () const |
| Returns a handle of the data contained in the memory primitive. On the CPU engine, this is a pointer to the allocated memory. More...
|
|
void | mkldnn::memory::set_data_handle (void *handle) const |
|
static mkldnn_data_type_t | mkldnn::memory::convert_to_c (data_type adata_type) |
|
static mkldnn_memory_format_t | mkldnn::memory::convert_to_c (format aformat) |
|
bool | mkldnn::operator== (mkldnn_data_type_t a, memory::data_type b) |
|
bool | mkldnn::operator!= (mkldnn_data_type_t a, memory::data_type b) |
|
bool | mkldnn::operator== (memory::data_type a, mkldnn_data_type_t b) |
|
bool | mkldnn::operator!= (memory::data_type a, mkldnn_data_type_t b) |
|
bool | mkldnn::operator== (mkldnn_memory_format_t a, memory::format b) |
|
bool | mkldnn::operator!= (mkldnn_memory_format_t a, memory::format b) |
|
bool | mkldnn::operator== (memory::format a, mkldnn_memory_format_t b) |
|
bool | mkldnn::operator!= (memory::format a, mkldnn_memory_format_t b) |
|
mkldnn_padding_kind_t | mkldnn::convert_to_c (padding_kind kind) |
|
mkldnn_prop_kind_t | mkldnn::convert_to_c (prop_kind kind) |
|
| mkldnn::reorder::primitive_desc::primitive_desc (const memory::primitive_desc &input, const memory::primitive_desc &output) |
|
engine | mkldnn::reorder::primitive_desc::get_engine () |
|
| mkldnn::reorder::reorder (const primitive_desc &aprimitive_desc, const primitive::at &input, const memory &output) |
|
| mkldnn::reorder::reorder (const primitive::at &input, const memory &output) |
|
| mkldnn::view::primitive_desc::primitive_desc (const memory::primitive_desc &input, memory::dims dims, memory::dims offsets) |
|
memory::primitive_desc | mkldnn::view::primitive_desc::dst_primitive_desc () const |
|
engine | mkldnn::view::primitive_desc::get_engine () |
|
| mkldnn::view::view (const primitive_desc &view_pd, primitive::at input) |
|
| mkldnn::view::view (memory input, memory::dims dims, memory::dims offsets) |
|
std::vector< const_mkldnn_primitive_desc_t > | mkldnn::concat::primitive_desc::cpp_to_c (std::vector< memory::primitive_desc > inputs) |
|
| mkldnn::concat::primitive_desc::primitive_desc (const memory::desc &output, int concat_dimension, std::vector< memory::primitive_desc > inputs) |
|
| mkldnn::concat::primitive_desc::primitive_desc (int concat_dimension, std::vector< memory::primitive_desc > inputs) |
|
memory::primitive_desc | mkldnn::concat::primitive_desc::dst_primitive_desc () const |
|
engine | mkldnn::concat::primitive_desc::get_engine () |
|
| mkldnn::concat::concat (const primitive_desc &concat_pd, std::vector< primitive::at > &inputs, const memory &output) |
|
std::vector< const_mkldnn_primitive_desc_t > | mkldnn::sum::primitive_desc::cpp_to_c (std::vector< memory::primitive_desc > inputs) |
|
| mkldnn::sum::primitive_desc::primitive_desc (const memory::desc &output, std::vector< double > scale, std::vector< memory::primitive_desc > inputs) |
|
| mkldnn::sum::primitive_desc::primitive_desc (std::vector< double > scale, std::vector< memory::primitive_desc > inputs) |
|
memory::primitive_desc | mkldnn::sum::primitive_desc::dst_primitive_desc () const |
|
engine | mkldnn::sum::primitive_desc::get_engine () |
|
| mkldnn::sum::sum (const primitive_desc &sum_pd, std::vector< primitive::at > &inputs, const memory &output) |
|
static mkldnn_stream_kind_t | mkldnn::stream::convert_to_c (kind akind) |
|
| mkldnn::stream::stream (kind akind) |
| Constructs a stream. More...
|
|
stream & | mkldnn::stream::submit (std::vector< primitive > primitives) |
| Submits a vector of primitives to a stream for computations. More...
|
|
bool | mkldnn::stream::wait (bool block=true) |
| Waits for all computations submitted to the stream to complete. More...
|
|
stream & | mkldnn::stream::rerun () |
|
| mkldnn::convolution_forward::desc::desc (prop_kind aprop_kind, algorithm aalgorithm, const memory::desc &src_desc, const memory::desc &weights_desc, const memory::desc &bias_desc, const memory::desc &dst_desc, const memory::dims strides, const memory::dims padding_l, const memory::dims padding_r, const padding_kind apadding_kind) |
|
| mkldnn::convolution_forward::desc::desc (prop_kind aprop_kind, algorithm aalgorithm, const memory::desc &src_desc, const memory::desc &weights_desc, const memory::desc &dst_desc, const memory::dims strides, const memory::dims padding_l, const memory::dims padding_r, const padding_kind apadding_kind) |
|
| mkldnn::convolution_forward::desc::desc (prop_kind aprop_kind, algorithm aalgorithm, const memory::desc &src_desc, const memory::desc &weights_desc, const memory::desc &bias_desc, const memory::desc &dst_desc, const memory::dims strides, const memory::dims dilates, const memory::dims padding_l, const memory::dims padding_r, const padding_kind apadding_kind) |
|
| mkldnn::convolution_forward::desc::desc (prop_kind aprop_kind, algorithm aalgorithm, const memory::desc &src_desc, const memory::desc &weights_desc, const memory::desc &dst_desc, const memory::dims strides, const memory::dims dilates, const memory::dims padding_l, const memory::dims padding_r, const padding_kind apadding_kind) |
|
| mkldnn::convolution_forward::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine) |
|
memory::primitive_desc | mkldnn::convolution_forward::primitive_desc::src_primitive_desc () const |
|
memory::primitive_desc | mkldnn::convolution_forward::primitive_desc::weights_primitive_desc () const |
|
memory::primitive_desc | mkldnn::convolution_forward::primitive_desc::bias_primitive_desc () const |
|
memory::primitive_desc | mkldnn::convolution_forward::primitive_desc::dst_primitive_desc () const |
|
engine | mkldnn::convolution_forward::primitive_desc::get_engine () |
|
| mkldnn::convolution_forward::convolution_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &weights, const primitive::at &bias, const memory &dst) |
|
| mkldnn::convolution_forward::convolution_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &weights, const memory &dst) |
|
| mkldnn::convolution_backward_data::desc::desc (algorithm aalgorithm, const memory::desc &diff_src_desc, const memory::desc &weights_desc, const memory::desc &diff_dst_desc, const memory::dims strides, const memory::dims padding_l, const memory::dims padding_r, const padding_kind apadding_kind) |
|
| mkldnn::convolution_backward_data::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine, const convolution_forward::primitive_desc &hint_fwd_primitive_desc) |
|
memory::primitive_desc | mkldnn::convolution_backward_data::primitive_desc::diff_src_primitive_desc () const |
|
memory::primitive_desc | mkldnn::convolution_backward_data::primitive_desc::weights_primitive_desc () const |
|
memory::primitive_desc | mkldnn::convolution_backward_data::primitive_desc::diff_dst_primitive_desc () const |
|
engine | mkldnn::convolution_backward_data::primitive_desc::get_engine () |
|
| mkldnn::convolution_backward_data::convolution_backward_data (const primitive_desc &aprimitive_desc, const primitive::at &diff_dst, const primitive::at &weights, const memory &diff_src) |
|
| mkldnn::convolution_backward_weights::desc::desc (algorithm aalgorithm, const memory::desc &src_desc, const memory::desc &diff_weights_desc, const memory::desc &diff_bias_desc, const memory::desc &diff_dst_desc, const memory::dims strides, const memory::dims padding_l, const memory::dims padding_r, const padding_kind apadding_kind) |
|
| mkldnn::convolution_backward_weights::desc::desc (algorithm aalgorithm, const memory::desc &src_desc, const memory::desc &diff_weights_desc, const memory::desc &diff_dst_desc, const memory::dims strides, const memory::dims padding_l, const memory::dims padding_r, const padding_kind apadding_kind) |
|
| mkldnn::convolution_backward_weights::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine, const convolution_forward::primitive_desc &hint_fwd_primitive_desc) |
|
memory::primitive_desc | mkldnn::convolution_backward_weights::primitive_desc::src_primitive_desc () const |
|
memory::primitive_desc | mkldnn::convolution_backward_weights::primitive_desc::diff_weights_primitive_desc () const |
|
memory::primitive_desc | mkldnn::convolution_backward_weights::primitive_desc::diff_bias_primitive_desc () const |
|
memory::primitive_desc | mkldnn::convolution_backward_weights::primitive_desc::diff_dst_primitive_desc () const |
|
engine | mkldnn::convolution_backward_weights::primitive_desc::get_engine () |
|
| mkldnn::convolution_backward_weights::convolution_backward_weights (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &diff_dst, const memory &diff_weights, const memory &diff_bias) |
|
| mkldnn::convolution_backward_weights::convolution_backward_weights (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &diff_dst, const memory &diff_weights) |
|
| mkldnn::convolution_relu_forward::desc::desc (const convolution_forward::desc conv_desc, const double negative_slope) |
|
| mkldnn::convolution_relu_forward::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine) |
|
engine | mkldnn::convolution_relu_forward::primitive_desc::get_engine () |
|
| mkldnn::convolution_relu_forward::convolution_relu_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &weights, const primitive::at &bias, const memory &dst) |
|
| mkldnn::convolution_relu_forward::convolution_relu_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &weights, const memory &dst) |
|
| mkldnn::lrn_forward::desc::desc (prop_kind aprop_kind, algorithm aalgorithm, const memory::desc &src_desc, int local_size, double alpha, double beta, double k) |
|
| mkldnn::lrn_forward::desc::desc (prop_kind aprop_kind, algorithm aalgorithm, const memory::desc &src_desc, int local_size, double alpha, double beta) |
|
| mkldnn::lrn_forward::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine) |
|
memory::primitive_desc | mkldnn::lrn_forward::primitive_desc::src_primitive_desc () const |
|
memory::primitive_desc | mkldnn::lrn_forward::primitive_desc::workspace_primitive_desc () const |
|
memory::primitive_desc | mkldnn::lrn_forward::primitive_desc::dst_primitive_desc () const |
|
engine | mkldnn::lrn_forward::primitive_desc::get_engine () |
|
| mkldnn::lrn_forward::lrn_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const memory &workspace, const memory &dst) |
|
| mkldnn::lrn_forward::lrn_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const memory &dst) |
|
| mkldnn::lrn_backward::desc::desc (algorithm aalgorithm, const memory::desc &data_desc, const memory::desc &diff_data_desc, int local_size, double alpha, double beta, double k) |
|
| mkldnn::lrn_backward::desc::desc (algorithm aalgorithm, const memory::desc &data_desc, const memory::desc &diff_data_desc, int local_size, double alpha, double beta) |
|
| mkldnn::lrn_backward::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine, const lrn_forward::primitive_desc &hint_fwd_primitive_desc) |
|
memory::primitive_desc | mkldnn::lrn_backward::primitive_desc::diff_src_primitive_desc () const |
|
memory::primitive_desc | mkldnn::lrn_backward::primitive_desc::workspace_primitive_desc () const |
|
memory::primitive_desc | mkldnn::lrn_backward::primitive_desc::diff_dst_primitive_desc () const |
|
engine | mkldnn::lrn_backward::primitive_desc::get_engine () |
|
| mkldnn::lrn_backward::lrn_backward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &diff_dst, const primitive::at &workspace, const memory &diff_src) |
|
| mkldnn::lrn_backward::lrn_backward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &diff_dst, const memory &diff_src) |
|
| mkldnn::pooling_forward::desc::desc (prop_kind aprop_kind, algorithm aalgorithm, const memory::desc &src_desc, const memory::desc &dst_desc, const memory::dims strides, const memory::dims kernel, const memory::dims padding_l, const memory::dims padding_r, const padding_kind apadding_kind) |
|
| mkldnn::pooling_forward::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine) |
|
memory::primitive_desc | mkldnn::pooling_forward::primitive_desc::workspace_primitive_desc () const |
|
memory::primitive_desc | mkldnn::pooling_forward::primitive_desc::dst_primitive_desc () const |
|
engine | mkldnn::pooling_forward::primitive_desc::get_engine () |
|
| mkldnn::pooling_forward::pooling_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const memory &dst) |
|
| mkldnn::pooling_forward::pooling_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const memory &dst, const memory &workspace) |
|
| mkldnn::pooling_backward::desc::desc (algorithm aalgorithm, const memory::desc &diff_src_desc, const memory::desc &diff_dst_desc, const memory::dims &strides, const memory::dims &kernel, const memory::dims &padding_l, const memory::dims &padding_r, const padding_kind apadding_kind) |
|
| mkldnn::pooling_backward::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine, const pooling_forward::primitive_desc &hint_fwd_primitive_desc) |
|
memory::primitive_desc | mkldnn::pooling_backward::primitive_desc::diff_src_primitive_desc () const |
|
engine | mkldnn::pooling_backward::primitive_desc::get_engine () |
|
| mkldnn::pooling_backward::pooling_backward (const primitive_desc &aprimitive_desc, const primitive::at &diff_dst, const memory &diff_src) |
|
| mkldnn::pooling_backward::pooling_backward (const primitive_desc &aprimitive_desc, const primitive::at &diff_dst, const primitive::at &workspace, const memory &diff_src) |
|
template<typename T > |
| mkldnn::eltwise_forward::desc::desc (prop_kind aprop_kind, algorithm alg_kind, const memory::desc &src_desc, T alpha=0, T beta=0) |
|
template<typename T > |
MKLDNN_DEPRECATED | mkldnn::eltwise_forward::desc::desc (prop_kind aprop_kind, const memory::desc &src_desc, T negative_slope) |
|
| mkldnn::eltwise_forward::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine) |
|
memory::primitive_desc | mkldnn::eltwise_forward::primitive_desc::dst_primitive_desc () const |
|
engine | mkldnn::eltwise_forward::primitive_desc::get_engine () |
|
| mkldnn::eltwise_forward::eltwise_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const memory &dst) |
|
template<typename T > |
| mkldnn::eltwise_backward::desc::desc (algorithm alg_kind, const memory::desc &diff_data_desc, const memory::desc &data_desc, T alpha=0, T beta=0) |
|
template<typename T > |
MKLDNN_DEPRECATED | mkldnn::eltwise_backward::desc::desc (const memory::desc &diff_data_desc, const memory::desc &data_desc, T negative_slope) |
|
| mkldnn::eltwise_backward::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine, const eltwise_forward::primitive_desc &hint_fwd_primitive_desc) |
|
memory::primitive_desc | mkldnn::eltwise_backward::primitive_desc::diff_src_primitive_desc () const |
|
engine | mkldnn::eltwise_backward::primitive_desc::get_engine () |
|
| mkldnn::eltwise_backward::eltwise_backward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &diff_dst, const memory &diff_src) |
|
| mkldnn::softmax_forward::desc::desc (prop_kind aprop_kind, const memory::desc &data_desc, int softmax_axis) |
|
| mkldnn::softmax_forward::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine) |
|
engine | mkldnn::softmax_forward::primitive_desc::get_engine () |
|
| mkldnn::softmax_forward::softmax_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const memory &dst) |
|
template<typename T > |
| mkldnn::batch_normalization_forward::desc::desc (prop_kind aprop_kind, const memory::desc &src_desc, T epsilon, unsigned flags) |
|
| mkldnn::batch_normalization_forward::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine) |
|
memory::primitive_desc | mkldnn::batch_normalization_forward::primitive_desc::weights_primitive_desc () const |
|
memory::primitive_desc | mkldnn::batch_normalization_forward::primitive_desc::mean_primitive_desc () const |
|
memory::primitive_desc | mkldnn::batch_normalization_forward::primitive_desc::variance_primitive_desc () const |
|
memory::primitive_desc | mkldnn::batch_normalization_forward::primitive_desc::dst_primitive_desc () const |
|
engine | mkldnn::batch_normalization_forward::primitive_desc::get_engine () |
|
| mkldnn::batch_normalization_forward::batch_normalization_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &mean, const primitive::at &variance, const primitive::at &weights, const memory &dst) |
|
| mkldnn::batch_normalization_forward::batch_normalization_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &mean, const primitive::at &variance, const memory &dst) |
|
| mkldnn::batch_normalization_forward::batch_normalization_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &weights, const memory &dst, const memory &mean, const memory &variance) |
|
| mkldnn::batch_normalization_forward::batch_normalization_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const memory &dst, const memory &mean, const memory &variance) |
|
| mkldnn::batch_normalization_forward::batch_normalization_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at &weights, const memory &dst) |
|
| mkldnn::batch_normalization_forward::batch_normalization_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const memory &dst) |
|
template<typename T > |
| mkldnn::batch_normalization_backward::desc::desc (prop_kind aprop_kind, const memory::desc &diff_data_desc, const memory::desc &data_desc, T epsilon, unsigned flags) |
|
| mkldnn::batch_normalization_backward::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine, const batch_normalization_forward::primitive_desc &hint_fwd_primitive_desc) |
|
memory::primitive_desc | mkldnn::batch_normalization_backward::primitive_desc::weights_primitive_desc () const |
|
memory::primitive_desc | mkldnn::batch_normalization_backward::primitive_desc::diff_weights_primitive_desc () const |
|
memory::primitive_desc | mkldnn::batch_normalization_backward::primitive_desc::mean_primitive_desc () const |
|
memory::primitive_desc | mkldnn::batch_normalization_backward::primitive_desc::variance_primitive_desc () const |
|
memory::primitive_desc | mkldnn::batch_normalization_backward::primitive_desc::dst_primitive_desc () const |
|
engine | mkldnn::batch_normalization_backward::primitive_desc::get_engine () |
|
| mkldnn::batch_normalization_backward::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) |
|
| mkldnn::batch_normalization_backward::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) |
|
| mkldnn::batch_normalization_backward::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) |
|
| mkldnn::inner_product_forward::desc::desc (prop_kind aprop_kind, const memory::desc &src_desc, const memory::desc &weights_desc, const memory::desc &bias_desc, const memory::desc &dst_desc) |
|
| mkldnn::inner_product_forward::desc::desc (prop_kind aprop_kind, const memory::desc &src_desc, const memory::desc &weights_desc, const memory::desc &dst_desc) |
|
| mkldnn::inner_product_forward::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine) |
|
memory::primitive_desc | mkldnn::inner_product_forward::primitive_desc::src_primitive_desc () const |
|
memory::primitive_desc | mkldnn::inner_product_forward::primitive_desc::weights_primitive_desc () const |
|
memory::primitive_desc | mkldnn::inner_product_forward::primitive_desc::bias_primitive_desc () const |
|
memory::primitive_desc | mkldnn::inner_product_forward::primitive_desc::dst_primitive_desc () const |
|
engine | mkldnn::inner_product_forward::primitive_desc::get_engine () |
|
| mkldnn::inner_product_forward::inner_product_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at weights, const primitive::at &bias, const memory &dst) |
|
| mkldnn::inner_product_forward::inner_product_forward (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at weights, const memory &dst) |
|
| mkldnn::inner_product_backward_data::desc::desc (const memory::desc &diff_src_desc, const memory::desc &weights_desc, const memory::desc &diff_dst_desc) |
|
| mkldnn::inner_product_backward_data::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine, const inner_product_forward::primitive_desc &hint_fwd_primitive_desc) |
|
memory::primitive_desc | mkldnn::inner_product_backward_data::primitive_desc::diff_dst_primitive_desc () const |
|
memory::primitive_desc | mkldnn::inner_product_backward_data::primitive_desc::weights_primitive_desc () const |
|
memory::primitive_desc | mkldnn::inner_product_backward_data::primitive_desc::diff_src_primitive_desc () const |
|
engine | mkldnn::inner_product_backward_data::primitive_desc::get_engine () |
|
| mkldnn::inner_product_backward_data::inner_product_backward_data (const primitive_desc &aprimitive_desc, const primitive::at &diff_dst, const primitive::at weights, const memory &diff_src) |
|
| mkldnn::inner_product_backward_weights::desc::desc (const memory::desc &src_desc, const memory::desc &diff_weights_desc, const memory::desc &diff_bias_desc, const memory::desc &diff_dst_desc) |
|
| mkldnn::inner_product_backward_weights::desc::desc (const memory::desc &src_desc, const memory::desc &diff_weights_desc, const memory::desc &diff_dst_desc) |
|
| mkldnn::inner_product_backward_weights::primitive_desc::primitive_desc (const desc &adesc, const engine &aengine, const inner_product_forward::primitive_desc &hint_fwd_primitive_desc) |
|
memory::primitive_desc | mkldnn::inner_product_backward_weights::primitive_desc::diff_dst_primitive_desc () const |
|
memory::primitive_desc | mkldnn::inner_product_backward_weights::primitive_desc::diff_weights_primitive_desc () const |
|
memory::primitive_desc | mkldnn::inner_product_backward_weights::primitive_desc::diff_bias_primitive_desc () const |
|
memory::primitive_desc | mkldnn::inner_product_backward_weights::primitive_desc::src_primitive_desc () const |
|
engine | mkldnn::inner_product_backward_weights::primitive_desc::get_engine () |
|
| mkldnn::inner_product_backward_weights::inner_product_backward_weights (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at diff_dst, const memory &diff_weights) |
|
| mkldnn::inner_product_backward_weights::inner_product_backward_weights (const primitive_desc &aprimitive_desc, const primitive::at &src, const primitive::at diff_dst, const memory &diff_weights, const memory &diff_bias) |
|