• J
    [KUNLUN] add op: maxpool_with_index (#49505) · f71f77e9
    jameszhang 提交于
    * [KUNLUN] add op: maxpool_with_index
    
    * use DeviceContext::Alloc() instead of DenseTensor::mutable_data()
    
    * fix file format
    
    * solve clip unittest failure
    
    * minor fix
    
    * Revert "solve clip unittest failure" since the issue is fixed
    in #49535
    
    This reverts commit 1127adc66e79afe35ac3c00bb34e6aaa7cd7d78b.
    
    * align with xdnn on the definition of mask in max_pool_with_index
    
    * minor
    f71f77e9
pool_kernel.cc 7.4 KB