1. 06 7月, 2021 1 次提交
    • Z
      Add gpu implementation of shuffle_batch_op (#33938) · c6b6ba1f
      Zeng Jinle 提交于
      * add gpu implementation of shuffle batch
      test=develop
      
      * add thrust cuda patches
      test=develop
      
      * fix macro guard
      
      * fix shuffle batch compile on windows/hip
      
      * fix hip compilation error
      
      * refine CMakeLists.txt
      
      * fix windows compile error
      
      * try to fix windows CI compilation error
      
      * fix windows compilation again
      
      * fix shuffle_batch op test on Windows
      c6b6ba1f
  2. 09 3月, 2020 1 次提交
  3. 27 12月, 2019 1 次提交
    • Z
      add shuffle batch op (#21674) · cee2ccb0
      zhoushiyu 提交于
      * add shuffle batch op, test=develop, test=document_preview
      
      * fix size_t conflict and check_output test=develop, test=document_preview
      
      * fix bug test=develop, test=document_preview
      
      * add unittest of shuffle_batch layer test=develop, test=document_preview
      
      * fix py coverage and op input type, test=develop, test=document_preview
      
      * fix py coverage, test=develop
      
      * fix en doc, test=develop
      
      * move to contrib test=develop
      
      * add unique_name test=develop
      
      * invoke shuffle_batch in contrib.layers test=develop
      cee2ccb0