PReLU |
您所在的位置:网站首页 › leaky_relu › PReLU |
Detailed Documentation
PReLU primitive A primitive to perform PReLU (leaky ReLU with trainable alpha parameter) See also: PReLU in developer guide Global Functions dnnl_status_t DNNL_API dnnl_prelu_forward_primitive_desc_create( dnnl_primitive_desc_t* primitive_desc, dnnl_engine_t engine, dnnl_prop_kind_t prop_kind, const_dnnl_memory_desc_t src_desc, const_dnnl_memory_desc_t weights_desc, const_dnnl_memory_desc_t dst_desc, const_dnnl_primitive_attr_t attr )Creates a primitive descriptor for a PReLU (leaky ReLU with trainable alpha parameter) forward propagation primitive. NOTE: weights descriptor is allowed to be initialized with dnnl_format_tag_any or with format_kind set to dnnl_format_kind_any.Parameters: primitive_desc Output primitive descriptor. engine Engine to use. prop_kind Propagation kind. Possible values are dnnl_forward_training and dnnl_forward_inference. src_desc Source memory descriptor. weights_desc Alpha parameters memory descriptor. dst_desc Destination memory descriptor. attr Primitive attributes (can be NULL). Returns: dnnl_success on success and a status describing the error otherwise. dnnl_status_t DNNL_API dnnl_prelu_backward_primitive_desc_create( dnnl_primitive_desc_t* primitive_desc, dnnl_engine_t engine, const_dnnl_memory_desc_t src_desc, const_dnnl_memory_desc_t weights_desc, const_dnnl_memory_desc_t diff_src_desc, const_dnnl_memory_desc_t diff_weights_desc, const_dnnl_memory_desc_t diff_dst_desc, const_dnnl_primitive_desc_t hint_fwd_pd, const_dnnl_primitive_attr_t attr )Creates a primitive descriptor for a PReLU (leaky ReLU with trainable alpha parameter) backward propagation primitive. NOTE: weights descriptor and diff_weights descriptor are allowed to be initialized with dnnl_format_tag_any or with format_kind set to dnnl_format_kind_any.Parameters: primitive_desc Output primitive descriptor. engine Engine to use. src_desc Source memory descriptor. weights_desc Alpha parameters memory descriptor. diff_src_desc Diff source memory descriptor. diff_weights_desc Diff alpha parameters memory descriptor. diff_dst_desc Diff destination memory descriptor. hint_fwd_pd Primitive descriptor for a respective forward propagation primitive. attr Primitive attributes (can be NULL). Returns: dnnl_success on success and a status describing the error otherwise. |
今日新闻 |
推荐新闻 |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |