1. 26 2月, 2022 1 次提交
    • F
      Move GumbelSoftmax OP to phi (#39873) · 581b2c64
      From00 提交于
      * Move GumbelSoftmax OP to phi
      
      * platform::errors -> phi::errors; GumbelSoftmaxGradInferMeta -> backend.h/cc
      
      * Use axis util in kernel impl
      
      * Remove namespace platform::errors
      
      * Use GetCPUEngine in Device Context
      581b2c64
  2. 15 9月, 2021 1 次提交
    • Y
      Add New OP: gumbel_softmax (#35506) · 18eda6c3
      YuanRisheng 提交于
      * Add New Op: gumbel_softmax
      
      * Add New Op: gumbel_softmax
      
      * Add New Op: gumbel_softmax (amend)
      
      * add __main__ function in unit test
      
      * fix bugs when test in windows ci
      
      * update en docs
      
      * delete reletive error in unit test
      
      * delete relative error in unit test
      
      * set hard=True in unit test
      18eda6c3
  3. 18 8月, 2020 1 次提交