tensorflow/core/util/port.h


1
2
3
4
5
6
7
8
9
10
11

#ifndef TENSORFLOW_UTIL_PORT_H_
#define TENSORFLOW_UTIL_PORT_H_

namespace tensorflow {

// Returns true if GOOGLE_CUDA is defined.
bool IsGoogleCudaEnabled();

}  // end namespace tensorflow

#endif  // TENSORFLOW_UTIL_PORT_H_