tensorflow/core/util/port.cc


1
2
3
4
5
6
7
8
9
10
11
12
13

#include "tensorflow/core/util/port.h"

namespace tensorflow {

bool IsGoogleCudaEnabled() {
#if GOOGLE_CUDA
  return true;
#else
  return false;
#endif
}

}  // end namespace tensorflow