- 03 5月, 2018 2 次提交
-
-
由 Tomasz Patejko 提交于
* 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
-
由 chengduo 提交于
* fix __shfl_down_sync_ of cross_entropy * use reduceSum * "fix ci"
-
- 02 5月, 2018 8 次提交
-
-
由 dzhwinter 提交于
* "fix double type error" * "fix ci"
-
由 Abhinav Arora 提交于
* Fix cpplint issues in sequence_padding * Fix typo in cu file * Fix dependencies of sequence_padding * Add include
-
由 chengduo 提交于
-
由 fengjiayi 提交于
-
由 qiaolongfei 提交于
-
由 qiaolongfei 提交于
-
由 qiaolongfei 提交于
-
由 Abhinav Arora 提交于
* Fix CPPLint issues in lstm_cpu_kernel.h * Fix CPPLint issues in math/math_function_test * Fix CPPLint issues in math/math_function_test * Fix CPPLint issues in math/concat.cc * Fix CPPLint issues in math/concat.cc * Fix CPPLint issues in math/concat.cc * Fix CPPLint issues in math/gru_cpu_kernel * Fix CPPLint issues in math/selected_rows_functor_test.cu * Fix compile error * Fix compile error
-
- 01 5月, 2018 1 次提交
-
-
由 Helin Wang 提交于
-
- 30 4月, 2018 1 次提交
-
-
由 dzhwinter 提交于
* "re-commit " * "picked up" * "fix ci" * "fix pdb hang up issue in cuda 9"
-
- 28 4月, 2018 5 次提交
-
-
由 ktlichkid 提交于
-
由 chengduoZH 提交于
-
由 Abhinav Arora 提交于
* Fix CPPLint errors * Fix CPPLint errors in sequence2batch * Fix compilation * Fix LSTM op and GRU op * Fix LSTMP op * Fix more cpplint errors in operators/math * Address Code review feedback
-
由 Kexin Zhao 提交于
-
由 Kexin Zhao 提交于
-
- 27 4月, 2018 9 次提交
-
-
由 qiaolongfei 提交于
-
由 qingqing01 提交于
-
由 Yancey1989 提交于
-
由 fengjiayi 提交于
-
由 ktlichkid 提交于
-
由 fengjiayi 提交于
-
由 dyning 提交于
* modify roi pool with lod and expose ROI Pooling into Python API * make lod code brief * make doc more clearly * make doc more clearly
-
由 Tomasz Patejko 提交于
* Some changes for Adam profiling * Adam optimization: initial Eigen optimization * Eigen Adam: flavour of adam can be chosen * Eigen Adam used for CPU by default. Plain Adam used for GPU * Eigen Adam: missing call to the Eigen functor added * Eigen Adam: revert changes in benchmarks * Eigen Adam: typo corrected
-
由 Abhinav Arora 提交于
* Fix CPPLint issue in test_engine * Fix CPPLint errors in operators/math * Fix compilation
-
- 26 4月, 2018 13 次提交
-
-
由 baiyf 提交于
-
由 fengjiayi 提交于
-
由 Abhinav Arora 提交于
* Fix more CPPLint issues * Fix more CPPLint issues * Fix more CPPLint issues * Fix CPPLint issues in operators/math and operators/reader
-
由 qiaolongfei 提交于
-
由 qiaolongfei 提交于
-
由 qiaolongfei 提交于
-
由 qiaolongfei 提交于
-
由 Yancey1989 提交于
-
由 fengjiayi 提交于
-
由 fengjiayi 提交于
-
由 fengjiayi 提交于
-
由 Siddharth Goyal 提交于
-
由 fengjiayi 提交于
-
- 25 4月, 2018 1 次提交
-
-
由 fengjiayi 提交于
-