• 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
gumbel_softmax_kernel_impl.h 3.3 KB