Caffe2 - C++ API
A deep learning, cross platform ML framework
Related Pages
Modules
Data Structures
Files
C++ API
Python API
GitHub
File List
Globals
torch
csrc
utils
cuda_enabled.h
1
#pragma once
2
3
namespace
torch
{
4
namespace
utils {
5
6
static
inline
bool
cuda_enabled() {
7
#ifdef USE_CUDA
8
return
true
;
9
#else
10
return
false
;
11
#endif
12
}
13
14
}
15
}
torch
Definition:
jit_type.h:17
Generated on Thu Mar 21 2019 13:06:24 for Caffe2 - C++ API by
1.8.11