Google Git
Sign in
android / platform / external / pytorch / 03cc0f587c / . / aten / src / ATen / cudnn
tree: 5330e18e7b52086297e51105fc3cf7e0b2fdd9c8 [path history] [tgz]
  1. AutocastRNN.cpp
  2. cudnn-wrapper.h
  3. Descriptors.cpp
  4. Descriptors.h
  5. Exceptions.h
  6. Handle.cpp
  7. Handle.h
  8. Handles.h
  9. README.md
  10. Types.cpp
  11. Types.h
  12. 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.