• B
    【PaddlePaddle Hackathon 2】24、为 Paddle 新增 nn.ChannelShuffle 组网 API (#40743) · bbaaf217
    BrilliantYuKaimin 提交于
    * Add infermeta for ChannelShuffle
    
    * Create channel_shuffle_grad_kernel.h
    
    * Create channel_shuffle_kernel.h
    
    * Create channel_shuffle_sig.cc
    
    * Create channel_shuffle_op.cc
    
    ChannelShuffle算子的描述
    
    * Create channel_shuffle_kernel_impl.h
    
    ChannelShuffle核函数的实现
    
    * Create channel_shuffle_grad_kernel_impl.h
    
    ChannelShuffle反向核函数的实现
    
    * Add kernel register of channel shuffle and grad
    
    注册ChannelShuffle及其反向的核函数
    
    * add nn.functional.channel_shuffle
    
    * add nn.ChannelShuffle
    
    * Create test_channel_shuffle.py
    
    * Update example of ChannelShuffle in vision.py
    
    * Update test_channel_shuffle.py
    
    * 修改channel_shuffle核函数的实现位置
    
    * 修正代码格式
    
    * 删除多余空格
    
    * 完善channel_shuffle的错误检查
    
    * Update unary.cc
    
    * Update channel_shuffle_op.cc
    
    * Update test_channel_shuffle.py
    
    * Update unary.cc
    
    * add channel_shuffle
    
    * Update test_channel_shuffle.py
    
    * Update vision.py
    
    * 调整代码格式
    
    * Update channel_shuffle_sig.cc
    
    * 更新ChannelShuffle的文档
    
    * 更新channel_shuffle的文档
    
    * remove ChannelShuffleOpArgumentMapping
    
    * add ChannelShuffleGradInferMeta
    
    * Update channel_shuffle_op.cc
    
    * 调整channel_shuffle及其梯度的核函数的位置
    bbaaf217
vision.py 16.6 KB