- 14 6月, 2018 1 次提交
-
-
由 whs 提交于
* Add mean_iou op. * Add unitest for mean iou op. * Add optional collections of confusion matrix and mean_iou. * Fix cuda kernel. * Refine code. 1. Merge computing in GPU to two kernel. 2. Use wrong array and correct array instead of confusion matrix. * Add python api and fix cuda kernel. * Fix comments. * Small fix. * Small fix.
-
- 13 6月, 2018 2 次提交
- 12 6月, 2018 2 次提交
-
-
由 Wu Yi 提交于
* enable docstring checker and fix comments * update * update by comments * fix build
-
由 qingqing01 提交于
* Add normalization operator. 1. Refine the raw norm_op and let it more general to support to normalize Tensor along any axis. 2. There is a bug in l2_normalize API, which lacks sqrt after `reduce_sum`. 3. Use norm_op to refine the l2_normalize API. 4. Fix bug in test_normalization_wrapper.py.
-
- 11 6月, 2018 2 次提交
- 08 6月, 2018 1 次提交
-
-
由 yuyang18 提交于
-
- 05 6月, 2018 3 次提交
- 01 6月, 2018 3 次提交
- 31 5月, 2018 3 次提交
- 30 5月, 2018 5 次提交
-
-
由 fengjiayi 提交于
-
由 fengjiayi 提交于
-
由 mozga-intel 提交于
-
由 fengjiayi 提交于
-
由 walloollaw 提交于
-
- 29 5月, 2018 2 次提交
-
-
由 qingqing01 提交于
-
由 fengjiayi 提交于
-
- 28 5月, 2018 1 次提交
-
-
由 fengjiayi 提交于
-
- 25 5月, 2018 1 次提交
-
-
由 baiyf 提交于
* unify UpsamplingBilinear2d interface specification * unify UpsamplingBilinear2d interface specification * fix name conventions * small fix about computation order
-
- 24 5月, 2018 1 次提交
-
-
由 baiyf 提交于
* Expose bilinear operator into Python API * delete unused lines * delete unused lines * fix typos * fix name arg and typos
-
- 23 5月, 2018 2 次提交
-
-
由 whs 提交于
* Enhance reduce op for multi dims. * Uncomment some unitest. * Uncomment unitest. * Remove unused code. * Fix infershape and python wrapper. * Add more examples. * Fix l2_normalize. * Fix normalization_wrapper. * Polish code. 1. Rename unitest function. 2. Rename const variable.
-
由 Yibing Liu 提交于
-
- 22 5月, 2018 2 次提交
- 17 5月, 2018 2 次提交
-
-
由 qingqing01 提交于
-
由 zhanghaichao 提交于
-
- 04 5月, 2018 1 次提交
-
-
由 Kexin Zhao 提交于
-
- 03 5月, 2018 1 次提交
-
-
由 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
-
- 27 4月, 2018 1 次提交
-
-
由 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
-
- 22 4月, 2018 1 次提交
-
-
由 whs 提交于
-
- 18 4月, 2018 1 次提交
-
-
由 qingqing01 提交于
* Add topk Python API. * Add unit test. * Remove the repeated API.
-
- 17 4月, 2018 2 次提交
-
-
由 Jacek Czaja 提交于
-
由 Jacek Czaja 提交于
- EPS added to softmax mkldnn primitive outcome is limited to training phase Fixes after review clang format fixes clang format fixes
-