1. 25 1月, 2019 2 次提交
    • B
      Adding ngraph_engine_op (#14948) · efce2567
      baojun 提交于
      * enable ngraph_engine_op
      test=develop
      
      * merge develop test=develop
      
      * avoid const_cast test=develop
      
      * rm ngraph_operator test=develop
      
      * Added TODO to move EnableNgraph test=develop
      
      * Add TODO to remove const_cast test=develop
      efce2567
    • C
      Revert conv transpose cudnn (#15514) · f8f91fb4
      chengduo 提交于
      * Revert "set constant for loss"
      
      This reverts commit 167933f678ccbb3563e949710279efe004a27731.
      
      * Revert "remove workspace_handle"
      test=develop
      This reverts commit b4aca8ede9e685bce1dfb1c59e63919f33432572.
      f8f91fb4
  2. 24 1月, 2019 5 次提交
    • Y
      Add the CUDA kernel for beam_search op (#15020) · 3008fa12
      Yiqun Liu 提交于
      * Refine the beam_search op and test.
      
      * A basic CUDA implementation of beam_search for small batch_size.
      
      * Implement CUDA kernel for beam_search_op.
      
      * Use multiple CUDA threads in the same block to select the top beam.
      
      * Update the python api of beam_search op.
      
      * Enable extend function in CPU kernel of beam_search op.
      
      * Unify the CUDA codes.
      test=develop
      
      * Unify the CPU kernel of beam_search op.
      
      * Ensure the seletced items of beam_search_op's CPU kernel sorted by scores.
      
      * Update the description of beam_search in API.spec.
      
      * Enable the use of CUDA kernel in beam_search op.
      
      * Exclude the beam_search's CUDA unittest when there is no CUDA gpu, and delete some debuging statements.
      test=develop
      
      * Follow comments.
      test=develop
      
      * Call the CPU kernel for beam_search op when batch_size > 4.
      test=develop
      
      * Remove the except of is_empty op in PrepareData.
      test=develop
      3008fa12
    • C
      Clean elementwise_op_function (#15502) · bf91d11e
      chengduo 提交于
      test=develop
      bf91d11e
    • T
      nce add check sample lables, test=develop (#15463) · 5cfc40de
      tangwei12 提交于
      * nce add check sample lables, test=develop
      5cfc40de
    • L
      Gpu memory monitoring (#15436) · 5d026a88
      liuwei1031 提交于
      * fix github issue 15267 test=develop
      
      * fix github issue 15267 test=develop
      
      * monitor the GPU usage during runtime
      
      * revert allocator_facade.cc change
      
      * comments update test=develop
      5d026a88
    • S
      lazy_allocator · 51227bd4
      sneaxiy 提交于
      test=develop
      51227bd4
  3. 23 1月, 2019 8 次提交
  4. 22 1月, 2019 9 次提交
  5. 21 1月, 2019 14 次提交
  6. 20 1月, 2019 1 次提交
  7. 19 1月, 2019 1 次提交