Google Git
Sign in
android / platform / external / pytorch / b7ebc00979 / . / aten / src / ATen / cudnn
tree: 0588740342bdecca3dc152dce1f7b3cda5a6a06b [path history] [tgz]
  1. cudnn-wrapper.h
  2. Descriptors.cpp
  3. Descriptors.h
  4. Exceptions.h
  5. Handle.cpp
  6. Handle.h
  7. Handles.h
  8. README.md
  9. Types.cpp
  10. Types.h
  11. Utils.h
aten/src/ATen/cudnn/README.md

All files living in this directory are written with the assumption that cuDNN is available, which means that these code are not guarded by #if AT_CUDNN_ENABLED(). Therefore, whenever you need to use definitions from here, please guard the #include<ATen/cudnn/*.h> and definition usages with #if AT_CUDNN_ENABLED() macro, e.g. native/cudnn/BatchNorm.cpp.