• vfdev's avatar
    Added rois shape check in C++ (#2794) · b8e93084
    vfdev authored
    * Added rois shape check in C++
    
    * Fixes code formatting
    
    * Remove accidental include
    
    * - Updated code according to the review
    - Replaced old AT_ASSERT/ERROR by new TORCH_CHECK
    b8e93084
ROIAlign_cpu.cpp 14.3 KB