1. 04 1月, 2020 1 次提交
  2. 07 5月, 2019 1 次提交
    • K
      Softmax_cross_entropy op add axis (#16806) · a71d8fdb
      Kaipeng Deng 提交于
      * add attr axis infershape. test=develop
      
      * add CUDA kernel. test=develop
      
      * fix unittest. test=develop
      
      * fix unittest for soft_label. test=develop
      
      * fix fp16 unittest. test=develop
      
      * remove comment code. test=develop
      
      * refine test for axis. test=develop
      
      * add python api. test=develop
      
      * fix doc. test=develop
      
      * fix fp16 unittest. test=develop
      
      * fix ngraph test. test=develop
      
      * fix ENFORCE for test_imperative_transformer. test=develop
      
      * fit for ngraph test. test=develop
      
      * fix after rebase develop. test=develop
      
      * fix doc. test=develop
      
      * fix API.spec. test=develop
      
      * fix test_layers. test=develop
      
      * fix format. test=develop
      a71d8fdb
  3. 11 9月, 2018 1 次提交
  4. 28 5月, 2018 1 次提交
  5. 12 2月, 2018 1 次提交
  6. 10 2月, 2018 2 次提交
  7. 26 12月, 2017 1 次提交
  8. 12 12月, 2017 1 次提交
    • Q
      Refine device context (#6433) · 61ec0b95
      QI JUN 提交于
      There are mainly following fixes:
      
      - take `DeviceContext` as the template parameter of math functors and OpKernel instead of `Place`
      - remove `eigen_device` interface in base class  `DeviceContext`
      - remove `GetEigenDevice` interface in `ExecutionContext` and base class `DeviceContext`
      - remove unused `platform::EigenDeviceConverter`
      - rename `REGISTER_OP_GPU_KERNEL` to `REGISTER_OP_CUDA_KERNEL`
      - rename `USE_GPU_ONLY_OP` to `USE_CUDA_ONLY_OP`
      61ec0b95
  9. 29 10月, 2017 1 次提交
    • Q
      support sparse output for lookup table grad op (#5145) · 008f40ce
      QI JUN 提交于
      * add sparse support for sum op
      
      * typo fix
      
      * fix gpu build error
      
      * fix unittest error
      
      * typo fix
      
      * infer var type and shape in op_test
      
      * follow comments
      
      * fix build error
      
      * bypass some unittests depend on NetOp
      
      * support sparse output for lookup table grad op
      
      * refine codes
      
      * fix gpu build error
      
      * fix lookup table grad gpu kernel
      
      * fix ci
      
      * fix ci
      
      * fix ci
      
      * fix bug in lookup_table_grad op
      
      * fix bug in test_word2vec
      
      * register double kernel for some operators
      
      * set is_sparse=True in test_word2vec
      
      * fix lookup table grad op CUDA kernel bug
      
      * disable test_modified_huber_loss_op temporarily
      
      * disable test_lstm_unit_op temporarily
      008f40ce
  10. 26 10月, 2017 1 次提交
  11. 29 9月, 2017 1 次提交
  12. 27 9月, 2017 1 次提交