struct dnnl::lbr_gru_forward::primitive_desc¶
Overview¶
Primitive descriptor for an LBR GRU forward propagation primitive. More…
#include <dnnl.hpp> struct primitive_desc: public dnnl::rnn_primitive_desc_base { // construction primitive_desc(); primitive_desc( const desc& adesc, const engine& aengine, bool allow_empty = false ); primitive_desc( const desc& adesc, const primitive_attr& attr, const engine& aengine, bool allow_empty = false ); primitive_desc(dnnl_primitive_desc_t pd); // methods memory::desc src_layer_desc() const; memory::desc src_iter_desc() const; memory::desc weights_layer_desc() const; memory::desc weights_iter_desc() const; memory::desc bias_desc() const; memory::desc dst_layer_desc() const; memory::desc dst_iter_desc() const; memory::desc workspace_desc() const; };
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; engine get_engine() const; const char* impl_info_str() const; memory::dim query_s64(query what) 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; bool next_impl(); primitive_desc_base(); primitive_desc_base(); primitive_desc_base(); primitive_desc_base(); memory::desc src_layer_desc() const; memory::desc src_iter_desc() const; memory::desc src_iter_c_desc() const; memory::desc weights_layer_desc() const; memory::desc weights_iter_desc() const; memory::desc weights_peephole_desc() const; memory::desc weights_projection_desc() const; memory::desc bias_desc() const; memory::desc dst_layer_desc() const; memory::desc dst_iter_desc() const; memory::desc dst_iter_c_desc() const; memory::desc diff_src_layer_desc() const; memory::desc diff_src_iter_desc() const; memory::desc diff_src_iter_c_desc() const; memory::desc diff_weights_layer_desc() const; memory::desc diff_weights_iter_desc() const; memory::desc diff_weights_peephole_desc() const; memory::desc diff_weights_projection_desc() const; memory::desc diff_bias_desc() const; memory::desc diff_dst_layer_desc() const; memory::desc diff_dst_iter_desc() const; memory::desc diff_dst_iter_c_desc() const; primitive_desc(); primitive_desc();
Detailed Documentation¶
Primitive descriptor for an LBR GRU forward propagation primitive.
Construction¶
primitive_desc()
Default constructor. Produces an empty object.
primitive_desc( const desc& adesc, const engine& aengine, bool allow_empty = false )
Constructs a primitive descriptor for a LBR GRU forward propagation primitive.
Parameters:
adesc |
Descriptor for a LBR GRU forward propagation primitive. |
aengine |
Engine to use. |
allow_empty |
A flag signifying whether construction is allowed to fail without throwing an exception. In this case an empty object will be produced. This flag is optional and defaults to false. |
primitive_desc( const desc& adesc, const primitive_attr& attr, const engine& aengine, bool allow_empty = false )
Constructs a primitive descriptor for a LBR GRU forward propagation primitive.
Parameters:
adesc |
Descriptor for a LBR GRU forward propagation primitive. |
attr |
Primitive attributes to use. |
aengine |
Engine to use. |
allow_empty |
A flag signifying whether construction is allowed to fail without throwing an exception. In this case an empty object will be produced. This flag is optional and defaults to false. |
primitive_desc(dnnl_primitive_desc_t pd)
Constructs a primitive descriptor for a LBR GRU forward propagation primitive from a C API primitive descriptor that must have a matching kind.
Parameters:
pd |
C API primitive descriptor for a LBR GRU forward propagation primitive. |
Methods¶
memory::desc src_layer_desc() const
Returns source layer memory descriptor.
Returns:
Source layer memory descriptor.
memory::desc src_iter_desc() const
Returns source iteration memory descriptor.
Returns:
Source iteration memory descriptor.
A zero memory descriptor if the primitive does not have a source iteration parameter.
memory::desc weights_layer_desc() const
Returns weights layer memory descriptor.
Returns:
Weights layer memory descriptor.
memory::desc weights_iter_desc() const
Returns weights iteration memory descriptor.
Returns:
Weights iteration memory descriptor.
memory::desc bias_desc() const
Returns bias memory descriptor.
Returns:
Bias memory descriptor.
A zero memory descriptor if the primitive does not have a bias parameter.
memory::desc dst_layer_desc() const
Returns destination layer memory descriptor.
Returns:
Destination layer memory descriptor.
memory::desc dst_iter_desc() const
Returns destination iteration memory descriptor.
Returns:
Destination iteration memory descriptor.
A zero memory descriptor if the primitive does not have a destination iteration parameter.
memory::desc workspace_desc() const
Returns the workspace memory descriptor.
Returns:
Workspace memory descriptor.
A zero memory descriptor if the primitive does not require workspace parameter.