• Z
    Add SparseCooTensor and SparseCsrTensor (#38906) · a7edb3f3
    zhangkaihuo 提交于
    * fix bug:
    1. atten: set the default value of attn_dropout_rate to None
    2. ffn: add activation parameter
    
    * for pure fp16
    
    * Add a SparseCsrTensor
    
    * remove unused functional
    
    * remove const
    
    * remove SetMemoberTensor
    
    * remove non_zero_nums_, the number of non zero elements of each batch can be obtained from the crows
    
    * SparseCooTensor
    
    * add SetMember
    
    * merge upstream; add SetMember
    
    * merge upstream
    
    * merge upstream; add newline at end of file
    
    * add newline at end of file
    
    * remove newline at end of file
    
    * remove newline at end of file
    
    * stash
    
    * user pten::framework::make_ddim
    
    * user pten::framework::make_ddim
    
    * merge upstream; use the latest mutable_data
    
    * merge upstream; use the latest mutable_data
    
    * return mutable dense tensor
    a7edb3f3
kernel_utils.h 14.3 KB