tensorflow/core/util/use_cudnn.h


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

// The utility to check whether we have Cudnn depenedency.

#ifndef TENSORFLOW_UTIL_USE_CUDNN_H_
#define TENSORFLOW_UTIL_USE_CUDNN_H_

namespace tensorflow {

bool CanUseCudnn();

}  // namespace tensorflow

#endif  // TENSORFLOW_UTIL_USE_CUDNN_H_