• Chris Sullivan's avatar
    [cuDNN:Part 1] minimal refactoring of op::reduce (#965) · 682f7b04
    Chris Sullivan authored
    * Refactored the cudnn reduce kernel to use the nGraph Shape -> cudnnTensorDescriptor cudnn helpers that the other kernels use.
    
    * Added cacheing to cudnn reduce op.
    
    * Adding back hashing call before returning primitive index to op::Reduce (bug fix).
    
    * [cuDNN:Part 2] Descriptor Creation/Destruction refactoring (#969)
    
    * Added a cuDNN descriptor factory which manages the construction and destruction of cuDNN descriptors.
    It correctly calls Create/Destroy based on the cuDNN descriptor type. Previously the Destroy functions were not being called.
    
    * Removed commented code and changed class to struct on cudnn_descriptor.
    
    * Added comments and updated a few variable names.
    
    * Clang compiled cuDNN kernels (those not part of CUDNNEmitter)
    now use the CUDNNDescriptor factory.
    682f7b04
Name
Last commit
Last update
.ci/travis/ubuntu Loading commit data...
cmake Loading commit data...
contrib/docker Loading commit data...
doc Loading commit data...
licenses Loading commit data...
maint Loading commit data...
python Loading commit data...
src Loading commit data...
test Loading commit data...
third-party Loading commit data...
.clang-format Loading commit data...
.gitignore Loading commit data...
.gitmodules Loading commit data...
.travis.yml Loading commit data...
CMakeLists.txt Loading commit data...
INSTALL.md Loading commit data...
LICENSE Loading commit data...
README.md Loading commit data...
VERSION.in Loading commit data...
changes.md Loading commit data...