Name | Date | Size | #Lines | LOC | ||
---|---|---|---|---|---|---|
.. | - | - | ||||
AutocastRNN.cpp | H A D | 25-Apr-2025 | 5.4 KiB | 129 | 98 | |
Descriptors.cpp | H A D | 25-Apr-2025 | 6.5 KiB | 196 | 169 | |
Descriptors.h | H A D | 25-Apr-2025 | 14.6 KiB | 409 | 316 | |
Exceptions.h | H A D | 25-Apr-2025 | 0 | 1 | 0 | |
Handle.cpp | H A D | 25-Apr-2025 | 1.7 KiB | 58 | 27 | |
Handle.h | H A D | 25-Apr-2025 | 193 | 10 | 6 | |
Handles.h | H A D | 25-Apr-2025 | 44 | 3 | 2 | |
README.md | H A D | 25-Apr-2025 | 390 | 5 | 4 | |
Types.cpp | H A D | 25-Apr-2025 | 1,016 | 39 | 32 | |
Types.h | H A D | 25-Apr-2025 | 310 | 15 | 9 | |
Utils.h | H A D | 25-Apr-2025 | 595 | 23 | 14 | |
cudnn-wrapper.h | H A D | 25-Apr-2025 | 540 | 17 | 12 |
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