1. 08 5月, 2018 1 次提交
    • Y
      Clean OpProtoAndCheckerMaker · 0e78cb69
      Yu Yang 提交于
      Do not use ctor
      
      * Reduce line of codes.
      * We can use virtual function for Maker now.
      * The implementation does not care what maker holds, it is easier to
      refactor later.
      0e78cb69
  2. 07 5月, 2018 8 次提交
  3. 06 5月, 2018 1 次提交
  4. 05 5月, 2018 8 次提交
  5. 04 5月, 2018 13 次提交
  6. 03 5月, 2018 9 次提交
    • L
      add relu converter and unit-test · beb12455
      Luo Tao 提交于
      beb12455
    • X
      Add timeline support for distributed training · 76d8b14b
      Xin Pan 提交于
      76d8b14b
    • Y
      Clean MatMul · 815d8884
      Yu Yang 提交于
      815d8884
    • C
      fix __shfl_down (#10362) · 62fed4cb
      chengduo 提交于
      62fed4cb
    • Y
      Fix compile · bc816035
      Yu Yang 提交于
      bc816035
    • C
      fix __shfl · e97c1a8c
      chengduoZH 提交于
      e97c1a8c
    • D
      Fix/fp64 (#10346) · f63ff90b
      dzhwinter 提交于
      * "fix double type error"
      
      * "fix ci"
      
      * "softmax fp64"
      
      * "fix momentum"
      
      * "fix ci"
      f63ff90b
    • Y
      Fix the bug when a input variable of op is dispensable. (#10268) · 6084af47
      Yiqun Liu 提交于
      * Fix the bug when a input variable of op is dispensable.
      
      * Add HasInputs/Outputs interfaces to OperatorBase.
      
      * Remove the unreferenced header file.
      6084af47
    • T
      MKLDNN implementation of batch normalization (#9904) · 4a497b82
      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
      4a497b82