• Z
    [Phi]Move hierarchical_sigmoid kernel to phi (#40553) · 64a7cbd3
    Zhang Zheng 提交于
    * first commit
    
    * fix compile error
    
    * support std::vector<std::srting>
    
    * fix
    
    * fix op support on GPU by chenweihang
    
    * pass test
    
    * infershape
    
    * add set_dtype
    
    * fix order
    
    * fix
    
    * unify the impl of dt and sr
    
    * fix
    64a7cbd3
hierarchical_sigmoid_op.cc 10.3 KB