• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1All files living in this directory are written with the assumption that cuDNN is available,
2which means that these code are not guarded by `#if AT_CUDNN_ENABLED()`. Therefore, whenever
3you need to use definitions from here, please guard the `#include<ATen/cudnn/*.h>` and
4definition usages with `#if AT_CUDNN_ENABLED()` macro, e.g. [native/cudnn/BatchNorm.cpp](native/cudnn/BatchNorm.cpp).
5