|
| primitive_desc ()=default |
| Default constructor. Produces an empty object.
|
|
| primitive_desc (const desc &adesc, const engine &aengine, bool allow_empty=false) |
| Constructs a primitive descriptor for a GRU forward propagation primitive. More...
|
|
| primitive_desc (const desc &adesc, const primitive_attr &attr, const engine &aengine, bool allow_empty=false) |
| Constructs a primitive descriptor for a GRU forward propagation primitive. More...
|
|
| primitive_desc (dnnl_primitive_desc_t pd) |
| Constructs a primitive descriptor for a GRU forward propagation primitive from a C API primitive descriptor that must have a matching kind. More...
|
|
memory::desc | src_layer_desc () const |
| Returns source layer memory descriptor. More...
|
|
memory::desc | src_iter_desc () const |
| Returns source iteration memory descriptor. More...
|
|
memory::desc | weights_layer_desc () const |
| Returns weights layer memory descriptor. More...
|
|
memory::desc | weights_iter_desc () const |
| Returns weights iteration memory descriptor. More...
|
|
memory::desc | bias_desc () const |
| Returns bias memory descriptor. More...
|
|
memory::desc | dst_layer_desc () const |
| Returns destination layer memory descriptor. More...
|
|
memory::desc | dst_iter_desc () const |
| Returns destination iteration memory descriptor. More...
|
|
memory::desc | workspace_desc () const |
| Returns the workspace memory descriptor. More...
|
|
| rnn_primitive_desc_base ()=default |
| Default constructor. Produces an empty object.
|
|
| rnn_primitive_desc_base (dnnl_primitive_desc_t pd, dnnl::prop_kind aprop_kind, dnnl::algorithm cell_kind) |
| Constructs an RNN primitive descriptor base from a C API primitive descriptor while checking that it actually describes the expected primitive by comparing propagation and primitive kinds. More...
|
|
memory::desc | src_layer_desc () const |
| Returns source layer memory descriptor. More...
|
|
memory::desc | src_iter_desc () const |
| Returns source iteration memory descriptor. More...
|
|
memory::desc | src_iter_c_desc () const |
| Returns source recurrent cell state memory descriptor. More...
|
|
memory::desc | weights_layer_desc () const |
| Returns weights layer memory descriptor. More...
|
|
memory::desc | weights_iter_desc () const |
| Returns weights iteration memory descriptor. More...
|
|
memory::desc | weights_peephole_desc () const |
| Returns weights peephole memory descriptor. More...
|
|
memory::desc | weights_projection_desc () const |
| Returns weights projection memory descriptor. More...
|
|
memory::desc | bias_desc () const |
| Returns bias memory descriptor. More...
|
|
memory::desc | dst_layer_desc () const |
| Returns destination layer memory descriptor. More...
|
|
memory::desc | dst_iter_desc () const |
| Returns destination iteration memory descriptor. More...
|
|
memory::desc | dst_iter_c_desc () const |
| Returns destination recurrent cell state memory descriptor. More...
|
|
memory::desc | diff_src_layer_desc () const |
| Returns diff source layer memory descriptor. More...
|
|
memory::desc | diff_src_iter_desc () const |
| Returns diff source iteration memory descriptor. More...
|
|
memory::desc | diff_src_iter_c_desc () const |
| Returns diff source recurrent cell state memory descriptor. More...
|
|
memory::desc | diff_weights_layer_desc () const |
| Returns diff weights layer memory descriptor. More...
|
|
memory::desc | diff_weights_iter_desc () const |
| Returns diff weights iteration memory descriptor. More...
|
|
memory::desc | diff_weights_peephole_desc () const |
| Returns diff weights peephole memory descriptor. More...
|
|
memory::desc | diff_weights_projection_desc () const |
| Returns diff weights projection memory descriptor. More...
|
|
memory::desc | diff_bias_desc () const |
| Returns diff bias memory descriptor. More...
|
|
memory::desc | diff_dst_layer_desc () const |
| Returns diff destination layer memory descriptor. More...
|
|
memory::desc | diff_dst_iter_desc () const |
| Returns diff destination iteration memory descriptor. More...
|
|
memory::desc | diff_dst_iter_c_desc () const |
| Returns diff destination recurrent cell state memory descriptor. More...
|
|
| primitive_desc (const_dnnl_op_desc_t desc, const primitive_attr *attr, const engine &aengine, const_dnnl_primitive_desc_t hint_fwd_pd, bool allow_empty=false) |
| Constructs a primitive descriptor. More...
|
|
| primitive_desc (const_dnnl_op_desc_t desc, const primitive_attr *attr, const engine &aengine, const_dnnl_primitive_desc_t hint_fwd_pd, bool allow_empty=false) |
| Constructs a primitive descriptor. More...
|
|
bool | next_impl () |
| Advances the primitive iterator to the next implementation. More...
|
|
| primitive_desc_base ()=default |
| Default constructor. Produces an empty object.
|
|
| primitive_desc_base (dnnl_primitive_desc_t pd, dnnl::primitive::kind prim_kind) |
| Constructs a primitive descriptor base object from a clone of a C API primitive descriptor after verifying that it is what the caller expects. More...
|
|
| primitive_desc_base (dnnl_primitive_desc_t pd, dnnl::primitive::kind prim_kind, dnnl::prop_kind aprop_kind) |
| Constructs a primitive descriptor base object from a clone of a C API primitive descriptor after verifying that it is what the caller expects. More...
|
|
| primitive_desc_base (dnnl_primitive_desc_t pd, dnnl::primitive::kind prim_kind, dnnl::prop_kind prop_kind1, dnnl::prop_kind prop_kind2) |
| Constructs a primitive descriptor base object from a clone of a C API primitive descriptor after verifying that it is what the caller expects. More...
|
|
| primitive_desc_base ()=default |
| Default constructor. Produces an empty object.
|
|
engine | get_engine () const |
| Returns the engine of the primitive descriptor. More...
|
|
const char * | impl_info_str () const |
| Returns implementation name. More...
|
|
memory::dim | query_s64 (query what) const |
| Returns a memory::dim value (same as int64_t). More...
|
|
memory::desc | query_md (query what, int idx=0) const |
| Returns a memory descriptor. More...
|
|
memory::desc | src_desc (int idx) const |
| Returns a source memory descriptor. More...
|
|
memory::desc | dst_desc (int idx) const |
| Returns a destination memory descriptor. More...
|
|
memory::desc | weights_desc (int idx) const |
| Returns a weights memory descriptor. More...
|
|
memory::desc | diff_src_desc (int idx) const |
| Returns a diff source memory descriptor. More...
|
|
memory::desc | diff_dst_desc (int idx) const |
| Returns a diff destination memory descriptor. More...
|
|
memory::desc | diff_weights_desc (int idx) const |
| Returns a diff weights memory descriptor. More...
|
|
memory::desc | src_desc () const |
| Returns a source memory descriptor. More...
|
|
memory::desc | dst_desc () const |
| Returns a destination memory descriptor. More...
|
|
memory::desc | weights_desc () const |
| Returns a weights memory descriptor. More...
|
|
memory::desc | diff_src_desc () const |
| Returns a diff source memory descriptor. More...
|
|
memory::desc | diff_dst_desc () const |
| Returns a diff destination memory descriptor. More...
|
|
memory::desc | diff_weights_desc () const |
| Returns a diff weights memory descriptor. More...
|
|
memory::desc | workspace_desc () const |
| Returns the workspace memory descriptor. More...
|
|
memory::desc | scratchpad_desc () const |
| Returns the scratchpad memory descriptor. More...
|
|
engine | scratchpad_engine () const |
| Returns the engine on which the scratchpad memory is located. More...
|
|
primitive_attr | get_primitive_attr () const |
| Returns the primitive attributes. More...
|
|
dnnl::primitive::kind | get_kind () const |
| Returns the kind of the primitive descriptor. More...
|
|
Public Member Functions inherited from dnnl::handle< dnnl_primitive_desc_t > |
bool | operator== (const handle< dnnl_primitive_desc_t, handle_traits< dnnl_primitive_desc_t > > &other) const |
| Equality operator. More...
|
|
bool | operator!= (const handle &other) const |
| Inequality operator. More...
|
|
| handle ()=default |
| Constructs an empty handle object. More...
|
|
| handle (const handle< dnnl_primitive_desc_t, handle_traits< dnnl_primitive_desc_t > > &)=default |
| Copy constructor.
|
|
| handle (handle< dnnl_primitive_desc_t, handle_traits< dnnl_primitive_desc_t > > &&)=default |
| Move constructor.
|
|
| handle (dnnl_primitive_desc_t t, bool weak=false) |
| Constructs a handle wrapper object from a C API handle. More...
|
|
handle< dnnl_primitive_desc_t, handle_traits< dnnl_primitive_desc_t > > & | operator= (const handle< dnnl_primitive_desc_t, handle_traits< dnnl_primitive_desc_t > > &)=default |
| Assignment operator.
|
|
handle< dnnl_primitive_desc_t, handle_traits< dnnl_primitive_desc_t > > & | operator= (handle< dnnl_primitive_desc_t, handle_traits< dnnl_primitive_desc_t > > &&)=default |
| Move assignment operator.
|
|
void | reset (dnnl_primitive_desc_t t, bool weak=false) |
| Resets the handle wrapper objects to wrap a new C API handle. More...
|
|
dnnl_primitive_desc_t | get (bool allow_empty=false) const |
| Returns the underlying C API handle. More...
|
|
| operator dnnl_primitive_desc_t () const |
| Converts a handle to the underlying C API handle type. More...
|
|
| operator bool () const |
| Checks whether the object is not empty. More...
|
|
Primitive descriptor for a GRU forward propagation primitive.