• Y
    Refine activation for GRU operator (#13275) · 87086b13
    Yihua Xu 提交于
    * Optimize GRU with AVX instruction
    
    * Clean code
    
    * Add the Unitest and fix the align issue
    
    * Remove the remanent part of the unitest part
    
    * Code clean
    
    * Fix the parameters length issue for fusion_gru to pass CI
    
    * Change the default type as float32
    87086b13
gru_cpu_kernel.h 17.8 KB