Searched refs:ADD_BOOL_CUDNN_FLAG (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | use_cudnn.cc | 29 #define ADD_BOOL_CUDNN_FLAG(func_name, flag_name, default_value) \ macro
78 ADD_BOOL_CUDNN_FLAG(CudnnUseAutotune, TF_CUDNN_USE_AUTOTUNE, true);
82 ADD_BOOL_CUDNN_FLAG(CudnnRnnUseAutotune, TF_CUDNN_RNN_USE_AUTOTUNE, true);
83 ADD_BOOL_CUDNN_FLAG(CudnnDisableConv1x1Optimization,
92 ADD_BOOL_CUDNN_FLAG(DebugCudnnRnn, TF_DEBUG_CUDNN_RNN, false);
97 ADD_BOOL_CUDNN_FLAG(DebugCudnnRnnUseTensorOps,
99 #undef ADD_BOOL_CUDNN_FLAG
|