• v-qjqs's avatar
    [Feature]: Support corner_pool related custom operators for onnxruntime in mmcv (#997) · db6b0542
    v-qjqs authored
    * supports for onnxruntime custom op `mmcv::MMCVTopPool`
    
    * supports for onnxruntime custom op `mmcv::MMCVCornerPool`, involving TopPool, BottomPool, LeftPool and RightPool
    
    * add unittest for corner_pool
    
    * supports mmcv::CornerPool without memcpy
    
    * add docs for mmcv::CornerPool
    
    * re-add docs for mmcv::CornerPool
    
    * fix output dtype doc
    
    * reformat
    
    * format with pre-commit
    
    * format
    
    * fix lint error, by using google clang-format style for c/c++
    db6b0542
test_onnx.py 17.9 KB