Dnnl_api_eltwise

Collaboration diagram for Dnnl_api_eltwise:

Classes
struct	dnnl_eltwise_desc_t
	A descriptor of a element-wise operation. More...

Functions
dnnl_status_t DNNL_API	dnnl_eltwise_forward_desc_init (dnnl_eltwise_desc_t *eltwise_desc, dnnl_prop_kind_t prop_kind, dnnl_alg_kind_t alg_kind, const dnnl_memory_desc_t *data_desc, float alpha, float beta)
dnnl_status_t DNNL_API	dnnl_eltwise_backward_desc_init (dnnl_eltwise_desc_t *eltwise_desc, dnnl_alg_kind_t alg_kind, const dnnl_memory_desc_t *diff_data_desc, const dnnl_memory_desc_t *data_desc, float alpha, float beta)

Detailed Description

Function Documentation

dnnl_eltwise_backward_desc_init()

dnnl_status_t DNNL_API dnnl_eltwise_backward_desc_init	(	dnnl_eltwise_desc_t *	eltwise_desc,
		dnnl_alg_kind_t	alg_kind,
		const dnnl_memory_desc_t *	diff_data_desc,
		const dnnl_memory_desc_t *	data_desc,
		float	alpha,
		float	beta
	)

Initializes a descriptor for eltwise backward propagation primitive.

Parameters

eltwise_desc	Output descriptor for an eltwise primitive.
alg_kind	Elementwise algorithm kind.
diff_data_desc	Diff source and diff destination memory descriptors.
data_desc	Source and destination memory descriptor.
alpha	The alpha parameter for the elementwise operation. Specific meaning depends on the algorithm.
beta	The beta parameter for the elementwise operation. Specific meaning depends on the algorithm.

Returns

dnnl_success on success and a status describing the error otherwise.

dnnl_eltwise_forward_desc_init()

dnnl_status_t DNNL_API dnnl_eltwise_forward_desc_init	(	dnnl_eltwise_desc_t *	eltwise_desc,
		dnnl_prop_kind_t	prop_kind,
		dnnl_alg_kind_t	alg_kind,
		const dnnl_memory_desc_t *	data_desc,
		float	alpha,
		float	beta
	)

Initializes a descriptor for eltwise forward propagation primitive.

Parameters

eltwise_desc	Output descriptor for an eltwise primitive.
prop_kind	Propagation kind. Possible values are dnnl_forward_training and dnnl_forward_inference.
alg_kind	Elementwise algorithm kind.
data_desc	Source and destination memory descriptor.
alpha	The alpha parameter for the elementwise operation. Specific meaning depends on the algorithm.
beta	The beta parameter for the elementwise operation. Specific meaning depends on the algorithm.

Returns

dnnl_success on success and a status describing the error otherwise.