Searched refs:ADD_BOOL_CUDNN_FLAG (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | use_cudnn.cc | 25 #define ADD_BOOL_CUDNN_FLAG(func_name, flag_name, default_value) \ macro
35 ADD_BOOL_CUDNN_FLAG(CanUseCudnn, TF_USE_CUDNN, true);
36 ADD_BOOL_CUDNN_FLAG(CudnnUseAutotune, TF_CUDNN_USE_AUTOTUNE, true);
40 ADD_BOOL_CUDNN_FLAG(CudnnRnnUseAutotune, TF_CUDNN_RNN_USE_AUTOTUNE, true);
41 ADD_BOOL_CUDNN_FLAG(CudnnDisableConv1x1Optimization,
50 ADD_BOOL_CUDNN_FLAG(DebugCudnnRnn, TF_DEBUG_CUDNN_RNN, false);
55 ADD_BOOL_CUDNN_FLAG(DebugCudnnRnnUseTensorOps,
57 #undef ADD_BOOL_CUDNN_FLAG
|