• Ayan Moitra's avatar
    Support ArgMin and ArgMax for NVGPU Backend (#1737) · 6f30b32b
    Ayan Moitra authored
    * Project initialization commit
    
    * Added unit tests for 3D tensors for argmax
    
    * Refactored reduce to be used by argmax argmin. argmax argmin still has some issues. WIP
    
    * [WIP]First working version of ArgMax ArgMin
    
    * added reduce buffer for the cudnn api calls
    
    * added reduce buffer for the cudnn api calls
    
    * Further modifications. Using rvalues to pass enums to build reduce method
    
    * more unit tests added
    
    * Incorporate Fenglei's comments
    
    * Incorporating Chris's first set of comments
    
    * small change to test file
    
    * Resolving clang issue that was causing argmin test to fail
    
    * Incorporate Chris's  comments
    
    * clang format issue
    6f30b32b
backend_test.in.cpp 421 KB