• Amy Zhuang's avatar
    Use Eigen kernel for Gather when axis is 0. (#3014) · ea64f5bf
    Amy Zhuang authored
    * Use Eigen kernel for Gather when axis is 0.
    
    * Fix style error.
    
    * Rename variables.
    
    Fix bugs.
    
    Use helper function.
    
    * Add one unit test.
    
    * Check if openmp is defined.
    
    * Change unit tests.
    
    * Use Eigen kernel for uint8_t type.
    
    Add one uint8_t unit test.
    
    Address PR feedback.
    
    * Update cpu emitter.
    ea64f5bf
cpu_kernels.hpp 9.31 KB