Paddle

Commit Graph

Author	SHA1	Message	Date
Tomasz Patejko	e26f51ce74	MKLDNN elementwis_add with default broadcast operations (#11544 ) * elementwise_add with bcast: Brian's implementation by Brian added, with default bcasts * elementwise_add with bcast: GetExpectedKernelType added to elementwise_op * elementwise_add with bcast: use_mkldnn attribute added * elementwise_add with bcast: changes after review and some formatting * elementwise_add with bcast: changes after style check * elementwise_add with bcast: changes after style check cont. * elementwise_add with bcast: MKLDNN unittests added * elementwise_add with bcast: original unittests with use_mkldnn flag * elementwise_add with bcast: handling of MKLDNN format corrected * elementwise_add with bcast: setting MKLDNN format turned into lambda * elementwise_add with bcast: MKDNN format setting turned into separate function * elementwise_add with bcast: condition for choosing MKLDNN simplified * elementwise_add with bcast: fix for MKLDNN format set incorrectly in bcasts * elementwise_add with bcast: changes in unittests for broadcasts * elementwise_add with bcast: fixes in unittests regarding dimensions * elementwise_add with bcast: bring back correct format setting in mklml grad path * elementwise_add with bcast: fixed compilation error	7 years ago
fengjiayi	e819f1b3a5	Add unittests to check channelwise add	7 years ago
Kexin Zhao	3da094fd7b	rearrange test	7 years ago
Kexin Zhao	4bf168b274	add fp16 kernel for elementwise add	7 years ago
xuwei06	e9b8ebf42c	Correctly handling variable with batch dimension for math ops. When the second argument contains batch dimension, the axis should be 0. Also makes elementwise ops more tolerant at handling tensors with trailing singular dimensions.	7 years ago
Luo Tao	b11956a0b5	move Fluid API code out of V2 API code	7 years ago

6 Commits (25ab4e0d6da6310e5c1cb73a064a1ea47bc74851)