Searched refs:use_tensor_ops (Results 1 – 2 of 2) sorted by relevance
1117 bool use_tensor_ops = in Create() local1121 if (use_tensor_ops && !allow_tensor_ops) { in Create()1128 use_tensor_ops ? CUDNN_TENSOR_OP_MATH : CUDNN_FMA_MATH; in Create()1131 use_tensor_ops ? CUDNN_TENSOR_OP_MATH : CUDNN_DEFAULT_MATH; in Create()2671 bool use_tensor_ops; in UseTensorOps() local2673 use_tensor_ops = desc->tensor_ops_enabled(); in UseTensorOps()2674 if (use_tensor_ops && !IsTensorMathEnabled(stream, type)) { in UseTensorOps()2679 use_tensor_ops = IsTensorMathEnabled(stream, type); in UseTensorOps()2681 return use_tensor_ops; in UseTensorOps()2700 bool use_tensor_ops; in GetCudnnConvolutionForwardAlgorithm() local[all …]
746 AlgorithmDesc(Index a, bool use_tensor_ops) { in AlgorithmDesc() argument748 proto_.set_math_type(use_tensor_ops ? AlgorithmProto::TENSOR_OP_MATH in AlgorithmDesc()