-
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
| Name |
Last commit
|
Last update |
|---|---|---|
| .ci | ||
| cmake | ||
| contrib/docker | ||
| doc | ||
| licenses | ||
| maint | ||
| python | ||
| src | ||
| test | ||
| .clang-format | ||
| .gitattributes | ||
| .gitignore | ||
| .gitmodules | ||
| .travis.yml | ||
| CMakeLists.txt | ||
| CODEOWNERS | ||
| CONTRIB.md | ||
| INSTALL.md | ||
| LICENSE | ||
| README.md | ||
| VERSION.in | ||
| changes.md |