1. 13 6月, 2019 1 次提交
    • W
      Fix gather and scatter op has same index bug cherry-pick from #17952 · 072347ff
      wawltor 提交于
      test=release/1.5
      cherry-pick from #17952
      The scatter op has a calc bug when the indices has same index, the scatter op use overwrite mode to calculate the same index, fix this bug by using the accumulate mode to calculate the same index.At the same time, the gather op has the same bug when the op calc the grad. And we use the lib of open-blas and eigen to optimize the time cost in accumulate mode.
      072347ff
  2. 25 5月, 2019 1 次提交
  3. 30 1月, 2019 1 次提交
    • Y
      Return parent_idx in beam_search op (#15520) · 16d54f7f
      Yiqun Liu 提交于
      * Refine beam_search_op to output an extra parent_idx tensor.
      test=develop
      
      * Fix the unittest test_beam_search_op.
      test=develop
      
      * Fix the merging mistake.
      test=develop
      16d54f7f
  4. 19 4月, 2018 1 次提交
  5. 12 2月, 2018 1 次提交
  6. 10 2月, 2018 2 次提交
  7. 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
  8. 04 10月, 2017 1 次提交
  9. 03 10月, 2017 1 次提交
  10. 29 9月, 2017 2 次提交
  11. 28 9月, 2017 1 次提交
  12. 22 8月, 2017 1 次提交
  13. 15 8月, 2017 1 次提交
  14. 07 8月, 2017 1 次提交
  15. 05 8月, 2017 1 次提交
  16. 02 8月, 2017 2 次提交
  17. 01 8月, 2017 1 次提交
    • Q
      use operator context and infer context (#3024) · 61ebacbc
      Qiao Longfei 提交于
      * use operator context
      
      * optimize code
      
      * update net infershape
      
      * update InferShape
      
      * disable override InferShape(scope) in OperatorBase
      
      * change InferShapeImpl to InferShape
      
      * add template to OperatorContext Input/Output
      
      * merge Input InputVar, Output OutputVar
      
      * change Inputs to MultiInput
      
      * fix conflict
      
      * fix MultiInput bugs and add unit test
      
      * rename KernelContext to ExecutionContext
      
      * clean code
      
      * change InferShape to protected
      
      * fix template bug
      
      * refine code
      
      * use InputVar instead of Input<Variable>
      
      * typo
      
      * optimize code
      61ebacbc
  18. 25 7月, 2017 1 次提交
  19. 19 7月, 2017 4 次提交
  20. 17 7月, 2017 3 次提交
  21. 14 7月, 2017 1 次提交
  22. 13 7月, 2017 2 次提交
    • Y
      Follow comments · 79b70c2d
      Yu Yang 提交于
      * Convert `op` --> `operators`
      * Remove AddType in OpProtoMaker, because type is part of registry.
      * Rename CPU_OR_GPU --> DEVICE_TYPE in registry macro.
      79b70c2d
    • Y
      Add a sample op, `add_op` · a0aaafe9
      Yu Yang 提交于
      * Refine register methods, make Op can get rid of whole-archieve
      * `USE_OP` before a op is used.
      * Add unittest for add_op.
      a0aaafe9