Caffe2 - C++ API
A deep learning, cross platform ML framework
Related Pages
Modules
Data Structures
Files
C++ API
Python API
GitHub
File List
Globals
aten
src
ATen
test
cuda_cudnn_test.cpp
1
#include <gtest/gtest.h>
2
3
#include <ATen/ATen.h>
4
#include <ATen/cuda/CUDAContext.h>
5
#include <ATen/cudnn/Descriptors.h>
6
#include <ATen/cudnn/Handle.h>
7
8
using namespace
at
;
9
using namespace
at::native
;
10
11
TEST(CUDNNTest, CUDNNTestCUDA) {
12
if
(!at::cuda::is_available())
return
;
13
manual_seed(123);
14
}
at::native
Definition:
Descriptors.cpp:9
at
Flush-To-Zero and Denormals-Are-Zero mode.
Definition:
AccumulateType.h:17
Generated on Thu Mar 21 2019 13:06:09 for Caffe2 - C++ API by
1.8.11