Searched refs:use_tensor_op_math (Results 1 – 1 of 1) sorted by relevance
725 void set_use_tensor_op_math(bool use_tensor_op_math) { in set_use_tensor_op_math() argument728 (use_tensor_op_math ? CUDNN_TENSOR_OP_MATH : CUDNN_FMA_MATH); in set_use_tensor_op_math()730 (use_tensor_op_math ? CUDNN_TENSOR_OP_MATH : CUDNN_DEFAULT_MATH); in set_use_tensor_op_math()