Visible to Intel only — GUID: GUID-F4732B0E-68FC-4DB6-8463-7977D7A3DBD8
Abs
AbsBackward
Add
AvgPool
AvgPoolBackward
BatchNormForwardTraining
BatchNormInference
BatchNormTrainingBackward
BiasAdd
BiasAddBackward
Clamp
ClampBackward
Concat
Convolution
ConvolutionBackwardData
ConvolutionBackwardWeights
ConvTranspose
ConvTransposeBackwardData
ConvTransposeBackwardWeights
Dequantize
Divide
DynamicDequantize
DynamicQuantize
Elu
EluBackward
End
Exp
GroupNorm
GELU
GELUBackward
HardSigmoid
HardSigmoidBackward
HardSwish
HardSwishBackward
Interpolate
InterpolateBackward
LayerNorm
LayerNormBackward
LeakyReLU
Log
LogSoftmax
LogSoftmaxBackward
MatMul
Maximum
MaxPool
MaxPoolBackward
Minimum
Mish
MishBackward
Multiply
Pow
PReLU
PReLUBackward
Quantize
Reciprocal
ReduceL1
ReduceL2
ReduceMax
ReduceMean
ReduceMin
ReduceProd
ReduceSum
ReLU
ReLUBackward
Reorder
Round
Select
Sigmoid
SigmoidBackward
SoftMax
SoftMaxBackward
SoftPlus
SoftPlusBackward
Sqrt
SqrtBackward
Square
SquaredDifference
StaticReshape
StaticTranspose
Subtract
Tanh
TanhBackward
TypeCast
Wildcard
enum dnnl_alg_kind_t
enum dnnl_normalization_flags_t
enum dnnl_primitive_kind_t
enum dnnl_prop_kind_t
enum dnnl_query_t
enum dnnl::normalization_flags
enum dnnl::query
struct dnnl_exec_arg_t
struct dnnl_primitive
struct dnnl_primitive_desc
struct dnnl::primitive
struct dnnl::primitive_desc
struct dnnl::primitive_desc_base
Overview
Detailed Documentation
enum dnnl_rnn_direction_t
enum dnnl_rnn_flags_t
enum dnnl::rnn_direction
enum dnnl::rnn_flags
struct dnnl::augru_backward
struct dnnl::augru_forward
struct dnnl::gru_backward
struct dnnl::gru_forward
struct dnnl::lbr_augru_backward
struct dnnl::lbr_augru_forward
struct dnnl::lbr_gru_backward
struct dnnl::lbr_gru_forward
struct dnnl::lstm_backward
struct dnnl::lstm_forward
struct dnnl::rnn_primitive_desc_base
struct dnnl::vanilla_rnn_backward
struct dnnl::vanilla_rnn_forward
Visible to Intel only — GUID: GUID-F4732B0E-68FC-4DB6-8463-7977D7A3DBD8
struct dnnl::primitive_desc_base
Overview
Base class for all primitive descriptors. More…
#include <dnnl.hpp>
struct primitive_desc_base: public dnnl::handle
{
// methods
engine get_engine() const;
const char* impl_info_str() const;
memory::dim query_s64(query what) const;
memory::dims get_strides() const;
memory::dims get_dilations() const;
memory::dims get_padding_l() const;
memory::dims get_padding_r() const;
float get_epsilon() const;
template <typename T = unsigned>
T get_flags() const;
dnnl::algorithm get_algorithm() const;
float get_alpha() const;
float get_beta() const;
int get_axis() const;
memory::dim get_local_size() const;
float get_k() const;
float get_p() const;
std::vector<float> get_factors() const;
dnnl::algorithm get_cell_kind() const;
dnnl::rnn_direction get_direction() const;
dnnl::algorithm get_activation_kind() const;
memory::dims get_kernel() const;
memory::dim get_group_size() const;
dnnl::prop_kind get_prop_kind() const;
memory::desc query_md(query what, int idx = 0) const;
memory::desc src_desc(int idx) const;
memory::desc dst_desc(int idx) const;
memory::desc weights_desc(int idx) const;
memory::desc diff_src_desc(int idx) const;
memory::desc diff_dst_desc(int idx) const;
memory::desc diff_weights_desc(int idx) const;
memory::desc src_desc() const;
memory::desc dst_desc() const;
memory::desc weights_desc() const;
memory::desc diff_src_desc() const;
memory::desc diff_dst_desc() const;
memory::desc diff_weights_desc() const;
memory::desc workspace_desc() const;
memory::desc scratchpad_desc() const;
engine scratchpad_engine() const;
primitive_attr get_primitive_attr() const;
dnnl::primitive::kind get_kind() const;
std::vector<uint8_t> get_cache_blob_id() const;
};
// direct descendants
struct primitive_desc;
struct primitive_desc;
struct primitive_desc;
struct primitive_desc;Inherited Members
public:
// methods
handle<T, traits>& operator = (const handle<T, traits>&);
handle<T, traits>& operator = (handle<T, traits>&&);
void reset(T t, bool weak = false);
T get(bool allow_empty = false) const;
operator T () const;
operator bool () const;
bool operator == (const handle<T, traits>& other) const;
bool operator != (const handle& other) const;Detailed Documentation
Base class for all primitive descriptors.
Methods
engine get_engine() constReturns the engine of the primitive descriptor.
Returns:
The engine of the primitive descriptor.
const char* impl_info_str() constReturns implementation name.
Returns:
The implementation name.
memory::dim query_s64(query what) constReturns a memory::dim value (same as int64_t).
Parameters:
what |
The value to query. |
Returns:
The result of the query.
memory::dims get_strides() constReturns strides.
Returns:
Strides.
An empty dnnl::memory::dims if the primitive does not have a strides parameter.
memory::dims get_dilations() constReturns dilations.
Returns:
Dilations.
An empty dnnl::memory::dims if the primitive does not have a dilations parameter.
memory::dims get_padding_l() constReturns a left padding.
Returns:
A left padding.
An empty dnnl::memory::dims if the primitive does not have a left padding parameter.
memory::dims get_padding_r() constReturns a right padding.
Returns:
A right padding.
An empty dnnl::memory::dims if the primitive does not have a right padding parameter.
float get_epsilon() constReturns an epsilon.
Returns:
An epsilon.
Zero if the primitive does not have an epsilon parameter.
template <typename T = unsigned>
T get_flags() constReturns flags.
Parameters:
T |
Flags enumeration type. |
Returns:
Flags.
Zero if the primitive does not have a flags parameter.
dnnl::algorithm get_algorithm() constReturns an algorithm kind.
Returns:
An algorithm kind.
dnnl::algorithm::undef if the primitive does not have an algorithm parameter.
float get_alpha() constReturns an alpha.
Returns:
An alpha.
Zero if the primitive does not have an alpha parameter.
float get_beta() constReturns a beta.
Returns:
A beta.
Zero if the primitive does not have a beta parameter.
int get_axis() constReturns an axis.
Returns:
An axis.
A negative number if the primitive does not have an axis parameter.
memory::dim get_local_size() constReturns an LRN local size parameter.
Returns:
An LRN local size parameter.
Zero if the primitive does not have an LRN local size parameter.
float get_k() constReturns an LRN K parameter.
Returns:
An LRN K parameter.
Zero if the primitive does not have an LRN K parameter.
float get_p() constReturns a reduction P parameter.
Returns:
A reduction P parameter.
Zero if the primitive does not have a reduction P parameter.
std::vector<float> get_factors() constReturns a resampling factors parameters.
Returns:
A vector of factors.
An empty vector if the primitive does not have a resampling factors parameter.
dnnl::algorithm get_cell_kind() constReturns an RNN cell kind parameter.
Returns:
An RNN cell kind parameter.
dnnl::algorithm::undef if the primitive does not have an RNN cell kind parameter.
dnnl::rnn_direction get_direction() constReturns an RNN direction parameter.
Returns:
An RNN direction parameter.
dnnl::rnn_direction::undef if the primitive does not have an RNN direction parameter.
dnnl::algorithm get_activation_kind() constReturns an RNN activation kind parameter.
Returns:
An RNN activation kind parameter.
dnnl::algorithm::undef if the primitive does not have an RNN activation kind parameter.
memory::dims get_kernel() constReturns a pooling kernel parameter.
Returns:
A pooling kernel parameter.
An empty dnnl::memory::dims if the primitive does not have a pooling kernel parameter.
memory::dim get_group_size() constReturns a group size parameter.
Returns:
A group size parameter.
Zero if the primitive does not have a group size parameter.
dnnl::prop_kind get_prop_kind() constReturns a propagation kind.
Returns:
A propagation kind.
dnnl::prop_kind::undef if the primitive does not have a propagation parameter.
memory::desc query_md(query what, int idx = 0) constReturns a memory descriptor.
NOTE:
There are also convenience methods dnnl::primitive_desc_base::src_desc(), dnnl::primitive_desc_base::dst_desc(), and others.
Parameters:
what |
The kind of parameter to query; can be dnnl::query::src_md, dnnl::query::dst_md, etc. |
idx |
Index of the parameter. For example, convolution bias can be queried with what = dnnl::query::weights_md and idx = 1. |
Returns:
The requested memory descriptor.
A zero memory descriptor if the primitive does not have a parameter of the specified kind or index.
memory::desc src_desc(int idx) constReturns a source memory descriptor.
Parameters:
idx |
Source index. |
Returns:
Source memory descriptor.
A zero memory descriptor if the primitive does not have a source parameter with index idx.
memory::desc dst_desc(int idx) constReturns a destination memory descriptor.
Parameters:
idx |
Destination index. |
Returns:
Destination memory descriptor.
A zero memory descriptor if the primitive does not have a destination parameter with index idx.
memory::desc weights_desc(int idx) constReturns a weights memory descriptor.
Parameters:
idx |
Weights index. |
Returns:
Weights memory descriptor.
A zero memory descriptor if the primitive does not have a weights parameter with index idx.
memory::desc diff_src_desc(int idx) constReturns a diff source memory descriptor.
Parameters:
idx |
Diff source index. |
Returns:
Diff source memory descriptor.
A zero memory descriptor if the primitive does not have a diff source parameter with index idx.
memory::desc diff_dst_desc(int idx) constReturns a diff destination memory descriptor.
Parameters:
idx |
Diff destination index. |
Returns:
Diff destination memory descriptor.
A zero memory descriptor if the primitive does not have a diff destination parameter with index idx.
memory::desc diff_weights_desc(int idx) constReturns a diff weights memory descriptor.
Parameters:
idx |
Diff weights index. |
Returns:
Diff weights memory descriptor.
A zero memory descriptor if the primitive does not have a diff weights parameter with index idx.
memory::desc src_desc() constReturns a source memory descriptor.
Returns:
Source memory descriptor.
A zero memory descriptor if the primitive does not have a source parameter.
memory::desc dst_desc() constReturns a destination memory descriptor.
Returns:
Destination memory descriptor.
A zero memory descriptor if the primitive does not have a destination parameter.
memory::desc weights_desc() constReturns a weights memory descriptor.
Returns:
Weights memory descriptor.
A zero memory descriptor if the primitive does not have a weights parameter.
memory::desc diff_src_desc() constReturns a diff source memory descriptor.
Returns:
Diff source memory descriptor.
A zero memory descriptor if the primitive does not have a diff source memory with.
memory::desc diff_dst_desc() constReturns a diff destination memory descriptor.
Returns:
Diff destination memory descriptor.
A zero memory descriptor if the primitive does not have a diff destination parameter.
memory::desc diff_weights_desc() constReturns a diff weights memory descriptor.
Returns:
Diff weights memory descriptor.
A zero memory descriptor if the primitive does not have a diff weights parameter.
memory::desc workspace_desc() constReturns the workspace memory descriptor.
Returns:
Workspace memory descriptor.
A zero memory descriptor if the primitive does not require workspace parameter.
memory::desc scratchpad_desc() constReturns the scratchpad memory descriptor.
Returns:
scratchpad memory descriptor.
A zero memory descriptor if the primitive does not require scratchpad parameter.
See also:
Primitive Attributes: Scratchpad
engine scratchpad_engine() constReturns the engine on which the scratchpad memory is located.
Returns:
The engine on which the scratchpad memory is located.
primitive_attr get_primitive_attr() constReturns the primitive attributes.
Returns:
The primitive attributes.
dnnl::primitive::kind get_kind() constReturns the kind of the primitive descriptor.
Returns:
The kind of the primitive descriptor.
std::vector<uint8_t> get_cache_blob_id() constReturns the cache blob ID of the primitive descriptor.
Returns:
The cache blob ID of the primitive descriptor.