11 Commits (25d44d40acfca5ed92dbc57fbaa2b01367a66f99)

Author	SHA1	Message	Date
Tao Luo	20120d9c97	Merge pull request #14608 from jczaja/prv-conv2d-transpose-mkldnn ... [MKL-DNN]conv2d transpose	6 years ago
qingqing01	731d45a39a	Enable BatchNorm to use global mean and variane during training (#14630) ... * Enable BatchNorm to use global mean and variane during training * Update doc and follow comments.	6 years ago
Jacek Czaja	fb24690a58	- conv2d transpose MKL-DNN ... test=develop - Added new header for MKLDNN reuse functionality - Extended conv2d_transpose GetExpectedKernelType for MKL-DNN supporrt - Buildable conv transpose mkldnn and conv mkldnn using conv template - Conv2d transpose roughlt implemented and buildable - Added modifications conv2d transpose MKLDNN unit tests - Fix to UT of conv2d transpose mkldnn op - Wrong type of MKLDNN primitive was chosen for conv2d transpose - HAcks for conv2d transpose - UT enalbed - Replaced copying loop with memcpy - Draft of passing lambda into AcquireMemory - Made reorder (IOHW->OIHW) to be called only once	6 years ago
Krzysztof Binias	fb4b4f8d57	Refactor code	7 years ago
Krzysztof Binias	50d3e6e96b	Reusing primitives for forward Batch Norm operator	7 years ago
mozga-intel	b8a04c2fa1	Duplicated code was moved to common function	7 years ago
mozga-intel	3b128337a1	The mkldnn batch norm supports other data format	7 years ago
pzelazko-intel	9a15c92317	bnorm+relu fuse for mkldnn (inference) (#11434) ... * bnorm+relu fuse for mkldnn * separate fuse_relu function * bug fix * proper while range in inference_transpiler * description fix * review fix * review fix * unit test for fwd batch norm+relu MKLDNN fuse	7 years ago
Luo Tao	5aac910b89	add url of cuda9.0_cudnn7_avx_mkl library	7 years ago
mozga-intel	7d5643562f	MKLDNN layout: Support for batch norm operator	7 years ago
Tomasz Patejko	4a497b826d	MKLDNN implementation of batch normalization (#9904) ... * Initial implementation of forward pass for MKLDNN batch norm * Added attributes for MKLDNN batch norm * MKLDNN batch norm forward pass passes unittest. Started working on backward * Backward pass for MKLDNN batch norm added * MKLDNN batch norm: scoring added to forward pass * MKLDNN batch norm: bias as input added; handling AnyLayout when kernel is looked up * MKLDNN batch norm: python unit tests added; mkldnn tests removed * MKLDNN batch norm: changes required by cpplint * MKLDNN batch norm: refactoring the operator * MKLDNN batch norm: saved variance inversed in backward pass for correct execution of MKLDNN unit tests * MKLDNN batch norm: refctoring, function for static/const cast to void* added * MKLDNN batch norm: remove AnyLayout from batch norm * MKLDNN batch norm: only NCHW format is supported. Unittests refactored * MKDNN batch norm: use_mkldnn added to attributes * MKLDNN batch norm: AnyLayout removed from unittest * MKLDNN batch norm: added CUDNN defines to batch norm * MKLDNN batch norm: undefined data_format variable corrected * MKLDNN batch norm: use_cudnn added, use of setUp method for configuring attributes * MKLDNN batch norm: added use_cudnn attribute to batch norm operator * MKLDNN batch norm: correcting batch norm unit tests for MKLDNN * MKLDNN batch norm: MKLDNN tests moved to another file; reverting changes for saved variance not being inverted * Change default layout to NCHW * MKLDNN batch norm: init_kernel_type method added to unit tests * MKLDNN batch norm: style changes * MKLDNN batch norm: unit tests refactored * MKLDNN batch norm: added use_mkldnn attribute to batch norm python interface	7 years ago