• Home
Name Date Size #Lines LOC

..--

AutocastRNN.cppD04-Jul-20255.4 KiB12998

Descriptors.cppD04-Jul-20256.5 KiB196169

Descriptors.hD04-Jul-202514.6 KiB409316

Exceptions.hD04-Jul-20250 10

Handle.cppD04-Jul-20251.7 KiB5827

Handle.hD04-Jul-2025193 106

Handles.hD04-Jul-202544 32

README.mdD04-Jul-2025390 54

Types.cppD04-Jul-20251,016 3932

Types.hD04-Jul-2025310 159

Utils.hD04-Jul-2025595 2314

cudnn-wrapper.hD04-Jul-2025540 1712

README.md

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