• Y
    [X86] Optimize gru and softmax (#2877) · 0a890050
    Yiqun Liu 提交于
    * Optimize softmax. When the input tensor is 2-D and axis is 1, there is no need to resize.
    
    * Optimize the gru, avoid calling Tensor::Slice.
    test=develop
    
    * Remove a std::vector in softmax.
    test=develop
    
    * Define CalculateSeqWidth to get the width of a sequence.
    test=develop
    0a890050
gru_compute.h 8.1 KB