• Implement FusedEmbeddingSeqPoolGradKernel with cblas_saxpy (#19770) · 93c85c93
    翟飞跃 提交于
    * Implement the operator with sprase matrix multiply
    
    * Update the URL of mklml library.
    
    test=develop
    
    * Disable MKLML implematation when using no-linux.
    
    test=develop
    
    * optimize bp with mkl sparse matrix
    test=develop
    
    * tmp add fused_emb_seq layer
    
    * Add the support of padding_idx attribute.
    
    test=develop
    
    * add padding_idx support
    test=develop
    
    * implement grad refer lego
    test=develop
    93c85c93
fused_embedding_seq_pool_op.cc 6.9 KB