|
| softmax_forward ()=default |
| Default constructor. Produces an empty object.
|
|
| softmax_forward (const primitive_desc &pd) |
| Constructs a softmax forward propagation primitive. More...
|
|
| primitive ()=default |
| Default constructor. Constructs an empty object.
|
|
| primitive (const_dnnl_primitive_desc_t c_pd) |
| Constructs a primitive from a C API primitive descriptor. More...
|
|
| primitive (const primitive_desc &pd) |
| Constructs a primitive from a primitive descriptor. More...
|
|
const_dnnl_primitive_desc_t | get_primitive_desc () const |
| Returns the C API primitive descriptor of the underlying C API primitive. More...
|
|
kind | get_kind () const |
| Returns the kind of the primitive. More...
|
|
void | execute (stream &stream, const std::unordered_map< int, memory > &args) const |
| Executes computations specified by the primitive in a specified stream. More...
|
|
| handle ()=default |
| Constructs an empty handle object. More...
|
|
| handle (const handle< T, traits > &)=default |
| Copy constructor.
|
|
| handle (handle< T, traits > &&)=default |
| Move constructor.
|
|
| handle (T t, bool weak=false) |
| Constructs a handle wrapper object from a C API handle. More...
|
|
Public Member Functions inherited from dnnl::handle< dnnl_primitive_t > |
bool | operator== (const handle< dnnl_primitive_t, handle_traits< dnnl_primitive_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_t, handle_traits< dnnl_primitive_t > > &)=default |
| Copy constructor.
|
|
| handle (handle< dnnl_primitive_t, handle_traits< dnnl_primitive_t > > &&)=default |
| Move constructor.
|
|
| handle (dnnl_primitive_t t, bool weak=false) |
| Constructs a handle wrapper object from a C API handle. More...
|
|
handle< dnnl_primitive_t, handle_traits< dnnl_primitive_t > > & | operator= (const handle< dnnl_primitive_t, handle_traits< dnnl_primitive_t > > &)=default |
| Assignment operator.
|
|
handle< dnnl_primitive_t, handle_traits< dnnl_primitive_t > > & | operator= (handle< dnnl_primitive_t, handle_traits< dnnl_primitive_t > > &&)=default |
| Move assignment operator.
|
|
void | reset (dnnl_primitive_t t, bool weak=false) |
| Resets the handle wrapper objects to wrap a new C API handle. More...
|
|
dnnl_primitive_t | get (bool allow_empty=false) const |
| Returns the underlying C API handle. More...
|
|
| operator dnnl_primitive_t () const |
| Converts a handle to the underlying C API handle type. More...
|
|
| operator bool () const |
| Checks whether the object is empty. More...
|
|
Softmax forward propagation primitive.